Linux下cuda9.1安装Pytorch和Torchvision|简记

📔 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.

  

 

  
1
  
2
  
3
  
4
  
5
  
6
  
7
 

意思是我的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

  

 

  
1
  
2
  
3
 

📘 其它博文推荐

• 首先，不论哪种 方式，都是可以的，但是都建议 配置 源 进行 下载加速
• 我自己平时 习惯会首先使用 pip 来进行 安装
• 推荐 conda 和 pip 的 加速配置方式如下
• conda 主要功能是在服务器中 搭建自己独立环境【不影响其他人】

相关博文推荐

• anaconda conda 切换为国内源 |windows 和 Linux配置方法
• linux和window设置 pip 镜像源 | 最实用的环境下载加速设置
• Linux下cuda10.0安装Pytorch和Torchvision|简记
• ubuntu18给当前用户安装cuda11.2 图文教程 | 配置cuDNN8.1

文章来源: positive.blog.csdn.net，作者：墨理学AI，版权归原作者所有，如需转载，请联系作者。

原文链接：positive.blog.csdn.net/article/details/107114853