linux cuda-11.8及cudnn安装
·
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
有版本信息输出则表示部署成功。
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)