Linux下cuda9.1安裝Pytorch和Torchvision|簡記

Linux下cuda9.1安裝Pytorch和Torchvision|簡記


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







發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章