AI

pytorch 1.9.1 cu102 버전 설치 방법

leonhong 2021. 11. 4. 13:07

파이토치 공식 홈페이지에서 버전 1.9.1버전이 안보임.(20211103)

 

카톡커뮤니티에서는 홈페이지 업데이트 중이 아니냐는 의견도 있었음.

 

어쨌든 내가 필요한 버전인 1.9.1과 cuda10.2 버전을 찾기 위해 

 

명령어 중 일부를 검색해서 비슷한 명령어를 찾아서 내가 원하는 형태로 변환했음.

 

일단 설치는 되었고 윈도우2016서버 클라우드 pc에서 9.2버전의 쿠다드라이버를 쓴다고 바꾸라고 하고 딥러닝 신경망을 안 돌아갔음. 드라이버 업데이트를 하려다가 혹시나 문제가 생길까바 그냥 다른 버전의 토치를 깜.

 

그래픽 드라이버를 업데이트하면 돌아갈듯 한데 일단 필요할 듯 하여 기록에 남겨둠.

 

pip3 install torch==1.9.1+cu102 torchvision==0.10.1+cu102 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html

'AI' 카테고리의 다른 글

이미지 학습을 손쉽게 도와주는 도구  (1) 2024.11.05
Python에서 Opencv 설치하기  (0) 2022.12.08
Pytorch 설치하기  (0) 2022.11.21
Numpy Index Slicing을 이용한 빠른 접근  (0) 2022.07.28