Ubuntu16.04下安装CUDA10.1

CUDA 10.1的安装流程和之前版本略有不同，主要是需要在高级选项里指定一下安装路径，其他基本一样。

原始文章在这里：
Ubuntu 16.04下cuda和cudnn安装

下载CUDA

安装CUDA

1）输入accept

2）默认是全部安装的，这里只选择CUDA Toolkit 10.1，其他都去掉勾选

3）选中CUDA Toolkit 10.1，按 A 跳转到 Advanced options，Enter 进入 Change Toolkit Install Path

4）输入安装路径，改成不需要 sudo 权限的路径即可，Enter 完成

5）去掉其他勾选项，选择 Done 返回上一层

6）选择 Install 即可

7）安装成功

注： cuda10.1 的安装最好使用 root 权限。

关于安装过程中的报错：

/usr/lib/x86_64-linux-gnu/libcublasLt.so.10.1.0.105 can't be opened
/usr/lib/x86_64-linux-gnu/libcublas.so.10.1.0.105 can't be opened
/usr/lib/x86_64-linux-gnu/libnvblas.so.10.1.0.105 can't be opened
 Completed with errors. See log at /tmp/cuda-installer.log for details.

也是未使用 root 权限导致安装过程被拒绝。

8）设置环境变量

安装完成后，设置环境变量：.bashrc

vi ~/.bashrc

在文件末尾加上：

export LD_LIBRARY_PATH=/data/zyy/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH
export PATH=/data/zyy/usr/local/cuda-10.1/bin:$PATH
export CUDA_HOME=/data/zyy/usr/local/cuda-10.1

:wq 保存退出后更新环境变量：

source ~/.bashrc

此时查看 cuda 版本：nvcc -V

可同时安装多版本cuda，需要使用哪个版本只需在环境变量中修改路径即可。

安装CUDNN

下载 CUDNN 需要登陆一下 nvidia 账号，已经支持微信登陆啦！

得到 cudnn-10.1-linux-x64-v7.5.0.56.tgz 文件。

1）解压到指定路径下

# 解压tgz文件到/data/zyy/usr/local/cuda-10.1目录下
tar -zvxf cudnn-9.2-linux-x64-v7.5.1.10.tgz -C /data/zyy/usr/local/cuda-10.1

得到一个 cuda 文件夹。

2）添加环境变量

将解压后的 cuda 文件夹中的 lib64 文件夹关联到环境变量中。

首先打开设置文件：

vi ~/.bashrc

在弹出的 .bashrc 文件的最后加入：

# 这里的 lib64 是解压的 tgz 文件中的 lib64 文件夹
export LD_LIBRARY_PATH=/data/zyy/usr/local/cuda10.1/cuda/lib64:$LD_LIBRARY_PATH

:wq 保存更改的文件后，更新一下环境变量：

source ~/.bashrc

[2021-11-15更新] 上面示意图中的 CUDA_HOME 后面不要加 :$CUDA_HOME，很容易出 bug，按照文字描述来

3）复制头文件

进入 /data/zyy/usr/local/cuda-10.1/cuda/include 路径下，执行操作：

# 将头文件cudnn.h复制到cuda-10.1安装目录下对应的include文件中
cp cudnn.h /data/zyy/usr/local/cuda-10.1/include/

然后重置 cudnn.h 文件的读写权限：

chmod a+r /data/zyy/usr/local/cuda-10.1/include/cudnn.h

4）检查安装是否成功

cudnn 8以下版本，查看 cudnn 版本：

cat /data/zyy/usr/local/cuda-10.1/include/cudnn.h | grep CUDNN_MAJOR -A 2

cudnn 8以上的版本信息放在了另一个文件 cudnn_version.h 中，应输入：

cat /data/zyy/usr/local/cuda-10.2/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2