ubuntu18.04安装cuda和cudnn
NVIDAI驱动安装,到官网下载
nvidia驱动下载
选择对应得型号进行下载安装即可
CUDA安装
选择runfile 格式进行下载
下载run文件时候使用.cn的域名能大大提高下载速度
安装时不需要安装驱动,因为上一步已经安装了
配置环境变量
vi /etc/profile
export PATH=//usr/local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.2/lib64$LD_LIBRARY_PATH
cuda 安装完成
CUDNN 安装
tar -xvf cudnn-linux-x86_64-xxx_cuda11.7-archive.tar.xz
cd cudnn-linux-x86_64-xx_cuda11.7-archive
sudo cp lib/* /usr/local/cuda-11.7/lib64/
sudo cp include/* /usr/local/cuda-11.7/include/
sudo chmod a+r /usr/local/cuda-11.7/lib64/*
sudo chmod a+r /usr/local/cuda-11.7/include/*
查看cuDNN版本
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2