Tensorflow 설치법

설치

 

tensorflow-gpu 버전을 설치 할려면 앞서 cuda, cudnn 설치 필요

[window] -> https://www.tensorflow.org/install/source_windows#tested_build_configurations

Windows의 소스에서 빌드  |  TensorFlow

로 가서 자기랑 맞는 버전의 gpu버전 확인 (만약 cuda 설치가 안되였다면 tensorflow-gpu가 아닌 tensorflow를 받을것)

 

먼저 깔려있는 cuda 버전 확인(난 cuda 11.0)

 

 

확인 후 위 사이트에서 쿠다 버전에 맞는 버전의 tensorflow-gpu 설치

 

pip install tensorflow-gpu==2.4.0 으로 설치

 

설치 확인

 

cuda 작동 확인

 

 

 

끝