下载CUDA11.8
逐步执行上图命令
编辑环境变量文件
sudo gedit ~/.bashrc
配置环境变量
export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
重启
重启
重启
重要的事情说三边 查看版本
nvcc -V
结果
安装cudnn
找到适合你的cudnn
https://developer.nvidia.com/rdp/cudnn-archive#a-collapse7415-10
安装cudnn
下载好以后解压(注意核对下载和解压名称是否一致)
tar -zxvf cudnn-11.8-linux-x64-v7.1.tgz
解压后进到里面:只在这一层级,右击打开终端,一条一条输入以下命令
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
查看装好了吗?
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
最后
重启
重启
重启 重要的事情说三边
输入
nvidia-smi