ubuntu查看cudnn是否安装成功_Ubuntu安装cuda和cudnn
安装CUDA1)安装cuda不需要注册账号,有两种格式 runfile与deb,网上说两种格式的都有,感觉如果之前安装驱动与deb对应的驱动版本一致,选择deb方便,否则选择runfile。按照官方给出的下载安装方法。注意:wget使用wget -c 命令,可以断点续传。2)runfile文件中实际包含了驱动,如果已经安装,注意取消驱动安装的勾选。3)环境变量添加:如果只针对本用户 sudo vi
- 安装CUDA
1)安装cuda不需要注册账号,有两种格式 runfile与deb,网上说两种格式的都有,感觉如果之前安装驱动与deb对应的驱动版本一致,选择deb方便,否则选择runfile。按照官方给出的下载安装方法。注意:wget使用wget -c 命令,可以断点续传。
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都可以。
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的软连接,所以拷贝到哪个下面都一样
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐



所有评论(0)