tensoflow 설치

sudo conda install tensorflow-gpu

NVIDIA CUDA, cudaTOOKIT , driver 설치

1. developer.nvidia.com/cuda-toolkit 접속
2. 공식 문서를통해 해당 OS버전에 맞는 명령어 선택후 설치 진행
   

Not found Module : tensorflow 와같은 에러 발생시 해결방법

pip install --upgrade tensorflow

Atribute Error : numpy 와같은 모듈끼리 버전이 맞지않는 종속성 에러 발생시 해결 방법

python3 -m pip install --upgrade h5py

Driver 설치방법

마지막에 "recommended" 붙어 있는 driver version을 download

sudo apt install nvidia-driver-525

설치후 재부팅 필수

---

설치 성공 확인 화면

Check Code

import tensorflow as tf

print("TensorFlow version:", tf.__version__)
print("CUDA version:", tf.sysconfig.get_build_info()['cuda_version'])
print("cuDNN version:", tf.sysconfig.get_build_info()['cudnn_version'])

print("Num GPUs Available: ", len(tf.config.experimental.list_physical_devices('GPU')))

with tf.device('/GPU:0'):
    print("GPU OK!")
    # 여기에 TensorFlow 코드 작성
    pass

위의 코드를 실행시 Result가 출력된다면 정상 설치 완료!

Result

Num GPUs Available:  0
GPU OK!