1. 安装CUDA

1)安装cuda不需要注册账号,有两种格式 runfile与deb,网上说两种格式的都有,感觉如果之前安装驱动与deb对应的驱动版本一致,选择deb方便,否则选择runfile。按照官方给出的下载安装方法。注意:wget使用wget -c 命令,可以断点续传。

dc0fe4d766df2174091bb3873c1da8d6.png

2)runfile文件中实际包含了驱动,如果已经安装,注意取消驱动安装的勾选。

3)环境变量添加:

如果只针对本用户 sudo vim ~/.bashrc

最后添加

export PATH="/usr/local/cuda-11.0/bin:$PATH"

export LD_LIBRARY_PATH="/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH"

export CUDA_HOME=/usr/local/cuda-11.0 (这个可能不需要)

然后执行

source ~/.bashrc

使得设置生效。

如果针对所有用户,可以sudo vim /etc/profile

最后添加

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64

export PATH=$PATH:/usr/local/cuda-11.0/bin

4)执行nvcc -V,查看是否安装成功。

2.安装cudnn,需要注册nvidia账号下载。根据安装cuda版本选择 第一项 或者 4+5+6都可以。

ce1a01837534be294d1b2e6c124bd28d.png

1)下载第一项的话,是一个solitairetheme8 后缀文件,执行

cp cudnn-11.0-linux-x64-v8.0.2.39.solitairetheme8 cudnn-11.0-linux-x64-v8.0.2.39.tgz cudnn-11.0-linux-x64-v8.0.2.39.solitairetheme8 cudnn-11.0-linux-x64-v8.0.2.39.tgz

tar -xvf cudnn-11.0-linux-x64-v8.0.2.39.solitairetheme8 cudnn-11.0-linux-x64-v8.0.2.39.tgz

结果应该和直接下载3个deb文件相同。

2)运行

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64
sudo chmod a+r /usr/local/cuda-11.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*

注:/usr/local/cuda 是 /usr/local/cuda-11.0的软连接,所以拷贝到哪个下面都一样

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐