프로그래밍

[Window10]RTX 3090 + CUDA 11.1 + cuDNN 8.0.5 + Python3.6

HamSung1231 2021. 12. 20. 14:22

연구실 컴퓨터를 RTX3090으로 바꾼 김에 최신 CUDA, Cudnn을 설치해보기로 했다.

그러나 2021-12-03 기준 CUDA 11.0&cuDNN 8.0.4과 CUDA 11.1&cuDNN 8.0.5를 해도 GPU가 가동되지 않았다.

 

결국 또 삽질의 시작....

 

CUDA 11.1

cuDNN 8.0.5
pytorch -> Stable (1.10) 

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

 

 

이렇게 하였는데도 되지 않았다. 문제는 결국 파이썬 버전 문제였던 것...

 

새로 가상 환경을 만들 때에 python = 3.6 버전으로 하니 GPU가 제대로 동작하였다...

 

삽질이 답이긴 한데, 제대로 된 호환 표를 제공해주었으면 하는 바람이...ㅜ