Linux下cuda9.1安裝Pytorch和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