2019年Ubuntu 18安装Pytorch、torchvision的最快捷方法

1、Ubuntu系统及下载源优化（如果已经更换国内源，请忽略此步）

#先备份原来的源列表文件 sudo cp /etc/apt/sources.list  /etc/apt/sourses.list.backup

#新建源列表文件 nano /etc/apt/sources.list #在该列表文件中加入源地址（推荐用阿里云源） deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse deb http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse deb http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse deb http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse deb-src http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse deb-src http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse deb-src http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse

#更新软件列表 sudo apt-get update

#更新软件（把已安装的软件与刚刷新的软件列表中的版本进行对比，如果发现已安装的软件版本太低，就会提示更新） sudo apt-get upgrade

2、直接使用apt下载安装NVIDIA驱动，这一步会自动帮你装好与驱动匹配的CUDA

#使用下面命令查看与显卡匹配的驱动

ubuntu-drivers devices

#然后，禁用掉开源驱动nouveaus（https://www.cnblogs.com/liangzp/p/9105294.html），选择你想要的驱动进行安装（会自动安装对应的CUDA）

sudo apt install nvidia-driver-430

（我装的是430版本的，因为我要装pytorch1.3 + torchvision0.4.1 + detectron2）

#安装好后，使用nvidia-smi查看驱动和CUDA信息（右上角）

3、直接Pip install一步到位（我的环境是：python3.6+pip19.2）

pip install torch torchvison

#这里会自动安装最新的Pytorch和torchvision（使用pip前请切换国内源）

然后就OK了。

 

这是我装了N次pytorch+torchvision后发现的最简单方法了。。。