Ubuntu18.04下安装CUDA 10.0和cuDNN 7.4

对应tensorflow版本：tensorflow-gpu1.15.2

CUDA安装

gcc --version检查gcc是否安装，如果未安装，使用sudo apt install build-essential命令安装build-essential软件包，其中包括gcc，g ++和make。

查看显卡信息：nvidia-smi

出现以上信息，说明已安装显卡驱动。其中的CUDA Version: 12.9是可以支持的最大CUDA版本。

注意：5090显卡（4060显卡也不支持）不支持这个cuda和cudnn版本，会报错：

failed to run cuBLAS routine: CUBLAS_STATUS_EXECUTION_FAILED
Internal: Blas GEMM launch failed : a.shape=(57, 512), b.shape=(512, 512)

 

下载CUDA Toolkit

CUDA Toolkit Archive | NVIDIA Developer

然后下载安装包

安装包下载到主目录后，运行以下命令

sudo sh cuda_10.0.130_410.48_linux.run

按空格键翻页到下图位置，然后输入accept回车，默认安装路径直接回车：

下载补丁，运行sudo sh cuda_10.0.130.1_linux.run命令安装

sudo vim ~/.bashrc打开.bashrc文件，在最后一行写入：

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

然后更新环境变量source ~/.bashrc

查看cuda版本：cat /usr/local/cuda-10.0/version.txt

nvcc --version

cuDNN安装

下载cuDNN

cuDNN Archive | NVIDIA Developer

 

点击下图链接，下载安装包

tar -zxvf cudnn-10.0-linux-x64-v7.4.2.24.tgz
对安装包解压，得到文件夹cuda

复制cuda文件夹下的文件到对应路径下：

（libcudnn.so和libcudnn.so.7是符号链接，这里用sudo cp -d单独复制可以保持原来大小外观。其实sudo cp cuda/lib64/* /usr/local/cuda-10.0/lib64/命令直接复制过去也可以）

sudo cp -d cuda/lib64/libcudnn.so /usr/local/cuda-10.0/lib64/
sudo cp -d cuda/lib64/libcudnn.so.7 /usr/local/cuda-10.0/lib64/
sudo cp cuda/lib64/libcudnn.so.7.4.2 /usr/local/cuda-10.0/lib64/
sudo cp cuda/lib64/libcudnn_static.a /usr/local/cuda-10.0/lib64/
sudo cp cuda/include/* /usr/local/cuda-10.0/include/

 查看cuDNN版本信息cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

参考链接

Ubuntu 20.04安装CUDA & CUDNN 手把手带你撸_ubuntu20.04安装cuda-CSDN博客

Ubuntu20.04下CUDA、cuDNN的详细安装与配置过程（图文）_ubuntu cudnn安装-CSDN博客