1. 安装CUDA

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

beed62dcb85f7e6de79a119ece1e8c65.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都可以。

60ad0d6689f90d043f0839cf2d4622fb.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

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐