Tensorflow-gpu 설치하기

랩실에서 2070을 줬는데 안쓰면 아까우니 tensorflow-gpu를 써봐야 하지 않겠나.

이분꺼 보면서 따라하며 설치했다.

[Setup] 딥러닝 개발 환경 구축 한방에 끝내기

내 환경

그래픽카드 : RTX 2070

쿠다 버전 : CUDA10.1

cuDNN 버전 : 8.0.5 ( 7.6인줄 알고 클릭했는데 8.0.5네..?)

 

해야하는거 

1. 아나콘다 설치

2. Visual Studio 2019 설치

3. CUDA 10.1설치

4. cuDNN v8.0.5설치

5. 가상환경 만들면서 python = 3.7 설치

conda create -n [가상환경이름] python=3.7

6. tensorflow-gpu =2.1설치 ( 처음에 2.3했는데 안되서 다운그레이드 했다 .)

conda install tensorflow-gpu=2.1

7. 주피터에 가상환경 커널 넘겨주고

import tensorflow
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

위 코드 실행해서 설치 제대로 됐는지, GPU 확인

굳.