win10 pytorch 환경 구축

509 단어
1. CUDA 버전 보기
nvcc --version

2. 청화원 연결을 구축한다. 그렇지 않으면 다운로드가 자꾸 중단된다.
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

3、pytorch 홈페이지에 로그인하여 CUDA버전에 해당하는pytorch를 다운로드
사이트 주소:
https://pytorch.org/get-started/previous-versions/

4, 값 부여 명령, 제거 -c, 그렇지 않으면 기본적으로pytorch 홈페이지에서 다운로드
conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 pytorch

좋은 웹페이지 즐겨찾기