pytorch model.cuda()는 오 랜 시간 동안 해결 되 었 습 니 다.

917 단어 pytorchmodel.cuda
해결 방법 중 하나:
pytorch 가 model.cuda()작업 을 하 는 데 오래 걸 리 고 GPU 의 속 도 를 의심 할 정도 로 길 어 지면 정상 이 아 닙 니 다.
만약 당신 이 사용 하 는 pytorch 버 전이 0.3.0 이 라면 0.3.1 로 업그레이드 하면 됩 니 다!
.cuda()로 딩 시간 이 긴 다른 해결 방법
방법 1:
pip install --upgrade --force-reinstall http://download.pytorch.org/whl/cu80/torch-0.2.0.post3-cp27-cp27mu-manylinux1_x86_64.whl
뒤에 이 건 홈 페이지 에서 자신 이 대응 하 는 python 과 cuda 버 전의 설치 패 키 지 를 찾 을 수 있 습 니 다.
홈 페이지 링크:

방법 2:
conda install pytorch torchvision cuda80 -c soumith
여 기 는 자신의 cuda 버 전에 따라 확정 할 수 있 습 니 다.cuda 버 전 을 보고 python 명령 줄 에 입력 하 십시오:

import torch
torch.version.cuda
됐다.
이상 은 개인 적 인 경험 이 므 로 여러분 에 게 참고 가 되 기 를 바 랍 니 다.여러분 들 도 저 희 를 많이 응원 해 주시 기 바 랍 니 다.

좋은 웹페이지 즐겨찾기