Ubuntu 18.04安装CUDA和cuDNN

目录

查看CUDA/CUDNN版本

通过conda在Ubuntu 18.04安装CUDA

通过NVIDIA官方提供的安装文件在Ubuntu 18.04安装CUDA

下载

执行安装程序

添加环境变量

测试是否成功

遇到的各种错误

Ubuntu 18.04 安装CUDNN

---

查看CUDA/CUDNN版本

cat /usr/local/cuda/version.txt

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

也可以通过nvidia-smi查看

通过conda在Ubuntu 18.04安装CUDA

conda install cudatoolkit=10.0

其中10.0为版本号，按需修改。

conda安装的CUDA Toolkit与Nvidia官方提供的CUDA Toolkit不一样

Nvidia 官方提供安装的CUDA Toolkit包含了进行 CUDA 相关程序开发的编译、调试等过程相关的所有组件

对于深度学习框架而言，在大多数需要使用 GPU 的情况中只需要使用 CUDA 的动态链接库支持程序的运行，不需要重新进行编译过程，这种情况用conda来安装是可以的

如果需要修改一些底层，涉及到CUDA相关程序开发时，则需要在Nvidia网站上下载完整的版本安装

通过NVIDIA官方提供的安装文件在Ubuntu 18.04安装CUDA

下载

CUDA Toolkit 11.6 Downloads | NVIDIA Developer

直接进去上述链接是最新版本

如果需要下载历史版本，网页上有个“Archive of Previous CUDA Releases”的链接

执行安装程序

比如下载的是cuda_10.0.130_410.48_linux.run，则执行

sh cuda_10.0.130_410.48_linux.run –no-opengl-libs

配置项参考（假设已经安装了显卡驱动）

添加环境变量

export PATH=/usr/local/cuda-10.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH

cuda的路径按需修改

测试是否成功

nvcc --version

遇到的各种错误

Missing gcc. gcc is required to continue.

安装gcc

sudo apt-get update

sudo apt-get install gcc

Ubuntu 18.04 安装CUDNN

前提是已安装CUDA

从网站上下载和解压

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

查看版本

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