파이선 텐서플로우 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값이 나와야 함