-
[Windows 10 + RTX3090] Pytorch 설치Setting & Error 2021. 8. 11. 01:20반응형
RTX3090을 사용하고 Windows10 os
1. CUDA, cuDNN 설치 된 환경이 필요
만약 CUDA, cuDNN 설치하지 않았다면 아래 사이트 참고해서 설치 진행
[Windows 10 + RTX3090] cudnn, cuda 재설치
1. 기존 NVIDIA 모두 삭제 1) 기존에 설치되어있던 nvidia 모두 삭제 2) "C:\Program Files (x86)" "C:\Program Files" 위 경로에 있는 NVIDIA 관련 파일 모두 삭제 3) "장치 관리자 -> 디스플레이어 어댑터 ->..
uding.tistory.com
2. pytorch 설치
PyTorch Korea User Group
(Unofficial) Korean user community for PyTorch which is an open source machine learning framework that accelerates the path from research prototyping to production deployment.
pytorch.kr
CUDA, cuDNN 버전에 맞게 다운로드
아나콘다 pytorch 가상환경 만들고 "conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge" 입력
설치 꽤 오래 걸림
반응형'Setting & Error' 카테고리의 다른 글