Linux下cuda9.1安装Pytorch和Torchvision|简记
PyToch和Torchvision对应版本 --官方安装命令
我自己尝试了 该链接中对应的许多版本,然而大都报错:
AssertionError:
The NVIDIA driver on your system is too old (found version 9010).
Please update your GPU driver by downloading and installing a new
version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install
a PyTorch version that has been compiled with your version
of the CUDA driver.
意思是我的cuda9.1版本不合适,让我 更新 cuda版本
解决方法:
如果是自己的服务器的话,我建议 直接 cuda版本 切换 到 9.0或者10.0 等其他版本,PyTorch 的支持会更好。坚持cuda9.1 不能变,那么 就要安装 torch1.1.0 和 torchvision0.3.0,这个对应版本亲测可用:
pip install torch==1.1.0
和
pip install torchvision==0.3.0