레이블이 #tensorflow인 게시물을 표시합니다. 모든 게시물 표시
레이블이 #tensorflow인 게시물을 표시합니다. 모든 게시물 표시

2021년 1월 14일 목요일

No need to set up cuDNN and CUDA in your ubuntu os for data science environment because..

visit this site I give you below

https://anaconda.org/anaconda/tensorflow-gpu

and follow the command.

conda install -c anaconda tensorflow-gpu 

This command will install all libraries and drivers into your conda environment.

Don't be stupid! ;)

2019년 8월 16일 금요일

텐서플로, 케라스 GPU 사용하는지 확인

https://rfriend.tistory.com/425


import numpy as np
import pandas as pd
import keras
import tensorflow as tf
from IPython.display import display
import PIL #Pillow

# How to check if the code is running on GPU or CPU?
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

tf.test.is_gpu_available(
    cuda_only=False,
    min_cuda_compute_capability=None
)
# Returns:
# True if a GPU device of the requested kind is available.