버전 맞추는데 꽤 애를 먹었는데 헷갈리지 않기 위해 정리해놓는다.
3000번대 시리즈는 cuda 11이상을 지원하고 이를 위해 다음과 같은 환경을 구성
Anaconda3-2020.02-Windows-x86_64
CUDA toolkit 11.1 # https://developer.nvidia.com/cuda-11.1.1-download-archive
cudnn v8.0.5 # https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111
cuDNN Archive
NVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.
developer.nvidia.com
anaconda 가상환경에서 다음과 같이 torch 버전을 깔아주었다.
pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
다음 세가지 테스트를 모두 마치면 성공이다.
torch.cuda.get_device_name(0)
torch.cuda.is_available()
torch.__version__