윈도우에서 딥러닝 개발환경 구축하기

14 Oct 2021 on Projects

Anaconda,CUDA,cuDNN,tensorflow,Pycharm

1. GPU드라이버

그래픽카드에 따라 호환되는 version이 다르므로 먼저 확인할 필요가 있음.
Compute capability확인 RTX30 시리즈는 compute capability가 8.6임

CUDA version확인 compute capability가 8.6은 CUDA 11.1이상의 버전이 필요함

cuDNN version확인 위에서 확인한 CUDA에 맞는 버전의 cuDNN설치.

cuDNN설치하고 압축을 풀어주는데, 압축 푸는 위치가 위에서 설치한 CUDA파일임! 덮어씌우면 됨

📌NVIDIA의 30시리즈는 tensorflow 2.5, CUDA 11.0이상, cuDNN 8.0이상

아나콘다 설치페이지 윈도우 버전으로 다운로드. 이때, 32비트 버전은 텐서플로가 설치되지 않으므로 64비트를 선택한다.

중간에 뜨는 이 창에서 두 개 모두 체크해준다.

RTX30시리즈는 tensorflow2.5와 호환되고 tensorflow2.5는 python9가 필요 가상환경을 만들어 준다.

(base) C:\Users\유다연> conda create --name tensorflow2.5 python=3.9

가상환경 이름은 tensorflow-gpu, 파이썬은 3.9버전 사용

가상환경을 만들었으면 활성화한다.

(base) C:\Users\유다연> activate tensorflow2.5

이제 이 가상환경에 텐서플로를 설치해준다.

(tensorflow2.5) C:\Users\유다연> conda install tensorflow-gpu=2.5

tensorflow 2.5버전으로 설치함

새 project를 만들 때 Base interpreter를 다음과 같이 설정

File > settings 에 들어가서 다음과 같이 설정

File > settings 에서 다음과 같이 +클릭 필요한 모듈 검색해서 설치해준다