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*