cuda 和 cudnn安装教程

cuda安装

cuda 版本查看:cat /usr/local/cuda/version.txt

CUDA下载页面:CUDA Toolkit Archive | NVIDIA Developer

450.57选了cuda-11.0。建议采用run或本地rpm安装,是会安装与之匹配的版本。远程rpm安装会安装最新版本。

下载界面有相应的yum源配置代码。

如果是本地rpm配置好后直接执行以下命令安装即可(会自动安装nvidia驱动):

sudo yum -y install cuda  //仅输入这一条命令即可,会自动下载其他cuda-driver和nvidia驱动等依赖

如果是采用run文件安装(可以选择不安装nvidia驱动):

运行cuda_10.0.130_410.48_linux.run,选择不安装推荐driver、不安装opengl,安装samples。生成目录:/usr/local/cuda-10.0/,samples目录/root,

创建软链接/usr/local/cuda/

显卡驱动与cuda版本对应关系:

卸载cuda(需要重启电脑):

Yum remove nvidia-*

yum remove cuda

rm -rf /usr/local/cuda-10.0

cudnn安装

cudnn 版本查看: cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

cudnn下载地址:(需要登录)

https://developer.nvidia.com/rdp/cudnn-download

选择和cuda对应版本的

比如下载cudnn-10.0-linux-x64-v7.6.5.32.tgz

解压后,cp cuda/include/cudnn.h /usr/local/cuda/include/

cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/

将上述文件权限设为所有人可读:

chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

猜你喜欢

转载自blog.csdn.net/qq_42152032/article/details/131342194