cuda查看
ls -l /usr/local | grep cuda
会显示安装的cuda版本,
cuda 11.1安装
https://developer.download.nvidia.cn/compute/cuda/11.1.1/local_installers/cuda_11.1.1_455.32.00_linux.run
记得accept后.—第一项选择去掉.这个是显卡驱动,如果没有可以勾选上
-[X] Driver!,把x去掉
cudnn安装
下载对应版本cuda版本的cudnn
https://developer.download.nvidia.cn/compute/cudnn/secure/8.9.6/local_installers/11.x/cudnn-linux-x86_64-8.9.6.50_cuda11-archive.tar.xz?fWkp_KM7o4cgZbdKpsH8z-C5PJpcjEOilFMokJ6uzgXTG5AjUArbGnmNM0mp6SUjK0pxjYS-n40kK_cSdi5jYDz6nULkYGK2neADjF6WbElZbi4TvkfSRwKajR-800zuyZ-DC24qUnlA16Wtjbug4r5HJzscc0riAvPa2o2cH_0GH5F3xFQnB_KB1ys6mR0ME1WBduhHytHcmfD0dHfy3EI=&t=eyJscyI6IndlYnNpdGUiLCJsc2QiOiJsaW5rLnpoaWh1LmNvbS8/dGFyZ2V0PWh0dHBzJTNBLy9kZXZlbG9wZXIubnZpZGlhLmNvbS9udmlkaWEtdGVuc29ycnQtZG93bmxvYWQifQ==
复制文件内容到指定目录下
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
修改权限
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
查看版本情况
cat /usr/local/**cuda-11.x**/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
修改不同的cuda
ln -s /usr/local/cuda-11.1 /usr/local/cuda