当直接按照PyTorch官网安装时,
pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html
输入:
python -c "import torch; print(torch.version.cuda)"
输出:
10.1
输入:
nvcc -V
输出:
10.0
所以,此时pytorch的cuda版本和系统的cuda版本是不匹配的。
解决方法:
pip install torch===1.4.0+cu100 torchvision===0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html
即可解决。