* cuda 설치
1. cuda 11.1 toolkit 다운로드
https://developer.nvidia.com/cuda-11.1.0-download-archive
CUDA Toolkit 11.1.0
Please Note: Due to an incompatibility issue, we advise users to defer updating to Linux Kernel 5.9+ until mid-November when an NVIDIA Linux GPU driver update with Kernel 5.9+ support is expected to be available. Select Target Platform Click on the green b
developer.nvidia.com
2. cudnn 다운로드 (cuda11.1 에 맞는 버젼으로)
Download cuDNN v8.0.5 (November 9th, 2020), for CUDA 11.1
cuDNN Archive
NVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.
developer.nvidia.com
3. cudnn 파일 복사
ex> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1
4. pytorch 설치
(ex> torch version 1.9.0 + cuda 11.1)
>> pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
https://download.pytorch.org/whl/torch_stable.html
download.pytorch.org
'Machine Learning' 카테고리의 다른 글
| torch 정상 동작 검사, 3080 그래픽카드 torch gpu 문제 해결 (0) | 2021.07.18 |
|---|
