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


1

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

📘 其它博文推荐


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

相关博文推荐


GitHub 加速计划 / li / linux-dash
10.39 K
1.2 K
下载
A beautiful web dashboard for Linux
最近提交(Master分支:2 个月前 )
186a802e added ecosystem file for PM2 4 年前
5def40a3 Add host customization support for the NodeJS version 4 年前
Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐