linux系统下cuda和cudnn的安装

0.cuda和cudnn各版本的官方地址

cuda:https://developer.nvidia.com/cuda-toolkit-archive
cudnn:https://developer.nvidia.com/rdp/cudnn-archive#a-collapse51b

1.安装cuda

sudo sh cuda_10.2.89_440.33.01_linux.run

1.1 修改安装cuda路径方法

# 更改临时存放位置使用--tmpdir
sudo sh cuda_11.0.3_450.51.06_linux.run --tmpdir=/home/usr/local
# 更改cuda toolkit 安装路径
<path>替换成安装位置的绝对路径
sudo sh cuda_10.0.130_410.48_linux.run --toolkitpath=<path> --toolkit -silent

2.安装cudnn

tar cudnn-10.1-linux-x64-v8.0.4.30.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.2/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

3.建立软链接

ln -s /usr/local/cuda-10.2  /usr/local/cuda

4.修改环境变量

sudo vim ~/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
export PATH=/usr/local/cuda/bin:$PATH
export CUDA_HOME=/usr/local/cuda

5.更新环境变量

source ~/.bashrc

6.查看cuda版本

cat  /usr/local/cuda/version.txt or  nvcc -V or nvcc -version

当以上两种命令获得的结果不一样时就需要检查nvcc

which nvcc  发现是/usr/bin/nvcc,而实际上nvcc是应该在/usr/local/cuda/bin/nvcc

因此需要所以,sudo vim /usr/bin/nvcc,把里面的内容"exec /usr/lib/nvidia-cuda-toolkit/bin/nvcc “ @ " " 改成 " e x e c / u s r / l o c a l / c u d a / b i n / n v c c " @"" 改成"exec /usr/local/cuda/bin/nvcc " @""改成"exec/usr/local/cuda/bin/nvcc"@”"

7.查看cudnn的版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

更新和安装NVIDIA 驱动程序
1:重启服务器

sudo reboot

2: 先卸载之前安装的显卡驱动程序

nvidia-installer --uninstall

3:给驱动run文件赋予执行权限

sudo chmod a+x NVIDIA-Linux-x86_64-410.93.run

4:安装驱动

sudo ./NVIDIA-Linux-x86_64-375.20.run –no-opengl-files
–no-opengl-files 只安装驱动文件,不安装OpenGL文件。这个参数最重要
–no-x-check 安装驱动时不检查X服务
–no-nouveau-check 安装驱动时不检查nouveau 
后面两个参数可不加。

对于选项依次选择:

Continue installation、No、No、Yes、OK
  • 0
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值