本文已参与「新人创作礼活动」，一起开启掘金创作之路。
安装Nvidia显卡驱动
sudo ubuntu-drivers autoinstall
安装完成后，使用命令nvidia-smi就可以查看到显卡的信息了，以960M为例:
安装CUDA10.0
1.下载CUDA Toolkit : Linux / x86_64 / Ubuntu / 18.04 /deb (local)
developer.nvidia.com/cuda-downlo…
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）
-
先解压 cudnn-10.1-linux-x64-v7.tgz 压缩包，然后将其中的内容复制到CUDA安装文件夹里面.
-
复制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*
