파이선 텐서플로우 gpu 설치

소프트웨어 요구사항

다음 NVIDIA® 소프트웨어가 시스템에 설치되어 있어야 합니다.

• NVIDIA® GPU 드라이버 - CUDA® 11.0에는 450.x 이상이 필요합니다.
• CUDA® Toolkit - TensorFlow는 CUDA® 11을 지원합니다(TensorFlow 2.4.0 이상).
• CUPTI는 CUDA® Toolkit과 함께 제공됩니다.
• cuDNN SDK 8.0.4(cuDNN 버전)

위에 설치 후

$ pip install tensorflow-gpu
$ SET PATH=C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0\\bin;%PATH%
$ SET PATH=C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0\\extras\\CUPTI\\lib64;%PATH%
$ SET PATH=C:\\Program Files\\NVIDIA GPU Computing Toolkit\\CUDA\\v11.0\\include;%PATH%
$ SET PATH=C:\\tools\\cuda\\bin;%PATH%

import tensorflow as tf
tf.__version__ # 버전 확인, '2.1.0'

from tensorflow.python.client import device_lib 
device_lib.list_local_devices() 
tf.test.is_gpu_available()
# cmd와 jupyter notebook cell에서 모두 True값이 나와야 함