1. CUDA Toolkit down (필자 test는rtx 3060, 11.3.0버전) https://developer.nvidia.com/cuda-toolkit-archive
CUDA Toolkit Archive
Previous releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production
developer.nvidia.com
2. cuDNN 설치(필자는 11.3버전) https://developer.nvidia.com/cudnn
3. 다운 받은 cuDNN압축풀어서 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3 경로에 넣어주기
4. 파이토치설치: pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
* 1번 CUDA설치가 되지 않으면 NVIDIA그래픽드라이버를 지우고 다시 설치
* 1번 CUDA를 설치하려는데 다른 설치프로그램이 있다고 하면 서비스에서 windows installer를 정지했다 재시작 후 재부팅하고 실행
5. cmd에서 파이썬 실행하고
>>> import torch
>>> torch.cuda.get_device_name(0)
했을 때 해당 그래픽카드가 검색되면 설치 완료
아래 참고 블로그에 자세한 설명 있음
'PyTorch' 카테고리의 다른 글
| [Pytorch] 기본기 (0) | 2023.06.11 |
|---|---|
| 설치된 파이토치 버전보기 (0) | 2022.08.30 |