tensorflow gpu 설치 - yarak001/machine_learning_common GitHub Wiki
- Windows
- GPU 확인
- nvidia-smi --query | fgrep 'Product Name'
- NVIDIA driver update
- GPU compute capability 확인
- GPU compute capability 호환 가능한 CUDA SDK 확인
- 원하는 CUDA SDK 설치
- CUDA SDK에 맞는 cuDNN설치
- 압축을 풀고 bin, lib, include file을 5번에서 설치한 CUDA SDK의 같은 directory에 복사
- Linux
from tensorflow.python.client import device_lib
device_lib.list_local_devices()
or
python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"