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

有版本信息输出则表示部署成功。

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐