windows 10에 cuda 10.0으로 pytorch 설치시 torch.cuda.is_available() false 문제 해결

https://pytorch.org/get-started/previous-versions/
여기서 제시하는 방식으로, (pip install torch==1.2.0 torchvision==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html)
cuda 10.0 버전으로 pytorch 설치시 only cpu 버전으로 깔리는 문제가 있다.

그 원인은 위에 명령어 실행 시 +cuda9.2버전으로 깔리는 것



정확한 위치를 지정해서 재설치하면 인식된다.

pip install https://download.pytorch.org/whl/cu100/torch-1.2.0-cp36-cp36m-win_amd64.whl



어이가 없어서 남겨놓음
확인 후 설치하시길

댓글

가장 많이 본 글