linux+cuda10.0
参考了官网的方法,总是报错,提示缺少cudatoolkit
最终参考了这篇博客 https://blog.csdn.net/tiandd12/article/details/102967888
手动下载了对应版本的torch与torchvision两个文件
地址:https://download.pytorch.org/whl/cu100/torch-1.3.0%2Bcu100-cp36-cp36m-linux_x86_64.whl
https://download.pytorch.org/whl/cu100/torchvision-0.4.1%2Bcu100-cp36-cp36m-linux_x86_64.whl
然后pip安装,查看
在cuda10.0下安装好的GPU版本的pytorch版本后会有“+cu100”,
测试GPU是否可用:
应该是安装成功了
后续有问题再回来修改。
来源:CSDN
作者:ycyang133
链接:https://blog.csdn.net/qq_41573611/article/details/103489482