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
됐다.이상 은 개인 적 인 경험 이 므 로 여러분 에 게 참고 가 되 기 를 바 랍 니 다.여러분 들 도 저 희 를 많이 응원 해 주시 기 바 랍 니 다.
이 내용에 흥미가 있습니까?
현재 기사가 여러분의 문제를 해결하지 못하는 경우 AI 엔진은 머신러닝 분석(스마트 모델이 방금 만들어져 부정확한 경우가 있을 수 있음)을 통해 가장 유사한 기사를 추천합니다:
정확도에서 스케일링의 영향데이터셋 스케일링은 데이터 전처리의 주요 단계 중 하나이며, 데이터 변수의 범위를 줄이기 위해 수행됩니다. 이미지와 관련하여 가능한 최소-최대 값 범위는 항상 0-255이며, 이는 255가 최대값임을 의미합니다. 따...
텍스트를 자유롭게 공유하거나 복사할 수 있습니다.하지만 이 문서의 URL은 참조 URL로 남겨 두십시오.
CC BY-SA 2.5, CC BY-SA 3.0 및 CC BY-SA 4.0에 따라 라이센스가 부여됩니다.