Ubuntu18.04+CUDA10.0+GTX1050Ti安装

安装Nvidia显卡驱动

sudo ubuntu-drivers autoinstall

安装完成后，使用命令nvidia-smi就可以查看到显卡的信息了，以960M为例:

安装CUDA10.0

1.下载CUDA Toolkit : Linux / x86_64 / Ubuntu / 18.04 /deb (local)

https://developer.nvidia.com/cuda-downloads

2.安装

sudo dpkg -i cuda-repo-ubuntu1804–10–0-local-10.0.130–410.48_1.0–1_amd64.deb
sudo apt-key add /var/cuda-repo-10–0-local-10.0.130–410.48/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

3.配置环境变量
打开sudo gedit ~/.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}}

然后设置环境变量和动态链接库，在命令行输入：

sudo gedit /etc/profile

在打开的文件末尾加入：

export PATH=/usr/local/cuda/bin:$PATH

保存之后，创建链接文件：

sudo gedit /etc/ld.so.conf.d/cuda.conf

在打开的文件中添加如下语句：

/usr/local/cuda/lib64

然后执行

sudo ldconfig1

使链接立即生效

4.测试一下：

cd ~/NVIDIA_CUDA-10.0_Samples/1_Utilities/deviceQuery
make
./deviceQuery

可以看到详细的信息，最后结果是PASS，说明CUDA安装成功。

安装cuDNN（针对18.04）

1. 先解压 cudnn-10.1-linux-x64-v7.tgz 压缩包，然后将其中的内容复制到CUDA安装文件夹里面.

2. 复制cuDNN内容到cuda相关文件夹内

• sudo cp cuda/include/cudnn.h /usr/local/cuda/include 注意，解压后的文件夹名称为cuda ,将对应文件复制到 /usr/local中的cuda内
• sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
• sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*