linux服务器上安装GPU版pytroch时遇到的问题

北城余情 提交于 2019-12-16 06:22:13

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是否可用:

应该是安装成功了

后续有问题再回来修改。 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!