Ubuntu16.04 + GeForce GTX 1080 + Python3.6 安装 CUDA8.0

torch.cuda.is_available()结果为false:

显卡驱动、cuda、pytorch三个版本对齐:
先从显卡驱动版本找到对应的cuda版本:https://blog.csdn.net/kellyroslyn/article/details/109668001
然后同时更新cuda和pytorch（还有torchvision）的版本:https://pytorch.org/get-started/previous-versions/

详细步骤：
1.使用nvidia-smi查询驱动版本

如图中Driver Version所示，该卡目前的驱动版本为384.130

2.安装cuda8.0
（1）参考链接：https://developer.nvidia.com/cuda-toolkit-archive，下载合适版本的cuda，我选择的是cuda8.0
（2）正式安装

sudo sh cuda_8.0.61_375.26_linux.run

除了Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 375.26?
(y)es/(n)o/(q)uit: n
输入n不安装nvidia图像驱动外，其余的都选择yes
（3）配置环境变量：在.bashrc中配置

sudo gedit ~/.bashrc

将以下内容写入到~/.bashrc尾部

export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3.安装cudnn
（1）开始的时候我下载的是三个.deb格式的包，后来发现咋都安不上，后使用.tar格式的安装包(下图中第一个)

（2）解压cuDNNU

tar -xzvf cudnn-9.0-linux-x64-v7.tgz

（3）复制下面的文件到你的cuda文件中

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64
sudo chmod a+r /usr/local/cuda-8.0/include/cudnn.h
/usr/local/cuda/lib64/libcudnn*

4.验证是否安装成功：
查看cuda版本信息:

nvcc -V

代码中进行验证：

import torch
print(torch.__version__)
print(torch.cuda.is_available())