Ubuntu16 Pytorch使用GPU：显卡驱动及CUDA安装超简易教程

声明：该教程仅提供简易安装方案，请谨慎参考，以免造成不可逆转的后果。

1 安装显卡驱动及CUDA

因为接下来使用的CUDA安装包已经包含显卡驱动，所以我们需要卸御已有的显卡驱动：

sudo apt-get remove --purge nvidia-*

然后官网下载CUDA驱动，官网链接：here。请注意，要下载.deb文件。当选择好文件后，官网会给出相应的Installation instructions，照着来没错。

接下来验证安装结果，报错说明没安装成功。

nvcc -V

2 安装PyTorch

到PyTorch官网（here）下载pytorch安装包或直接使用命令安装，请注意科学上网：），然后检查显卡是否可用：

python

>>> import pytorch
>>> print(torch.cuda.is_available())

返回 true说明安装成功。