cuda安装:
查看ubuntu版本:
lsb_release -a
cuda下载网址:
https://developer.nvidia.com/cuda-toolkit-archive
选择runfile方式:
执行以上指令后,根据终端提示,取消选择显卡驱动的安装即可。
将cuda-11.8添加到bashrc:
export PATH="/usr/local/cuda-11.8/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH"
cudnn安装:
下载网址:https://developer.nvidia.com/rdp/cudnn-archive
本文下载的cudnn版本为8.9.5。
tar -xvJf <cudnn文件.tar.xz>
cp ./include/cudnn*.h /usr/local/cuda-11.8/include/
cp ./lib/libcudnn* /usr/local/cuda-11.8/lib64/
chmod a+r /usr/local/cuda-11.8/include/cudnn*.h
chmod a+r /usr/local/cuda-11.8/lib64/libcudnn*
验证是否部署成功:
cat /usr/local/cuda-11.8/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
有版本信息输出则表示部署成功。