Ubuntu安装cuda+cudnn保姆级教程记录
发布网友
发布时间:2024-10-01 01:53
我来回答
共1个回答
热心网友
时间:2024-10-24 04:08
1. 查看驱动版本
此处显示的Driver Version是驱动的实际版本,而CUDA Version是对应的cuda最高版本,以上图为例,对应现有的驱动版本咱们应该安装12.1以下版本的cuda。
2. 安装cuda
官方链接: CUDA Toolkit - Free Tools and Training
点击download now,将会跳转至最新版本的安装界面
如果这不是你想要的版本,则滑动滚轮至最下方,找到Archive of Previous CUDA Releases,点击进入选择你想要安装的cuda版本,并选择服务器类型,根据下方显示的Installation Instructions安装cuda即可。
若运行以下命令,能显示图示信息,那么恭喜你,安装成功啦!
用vim命令打开环境配置文件.bashrc
输入i进入编辑模式,在文件的最后输入以下两行内容添加环境变量,esc退出编辑模式,并输入:wq进行保存。此处/usr/local/cuda-11.8/替换成自己的安装路径即可,若不清楚,可以运行whereis cuda查看。
重新加载环境配置文件
搞定!
我安装的时候(下载安装包.run文件的时候)不知道是网络问题还是什么,报段错误了,运维也没帮我解决,我是在别的机器下载的安装包上传到服务器,然后再运行sh安装的,一个笨方法,仅供参考哈哈哈哈哈哈。
3. 安装cudnn
官方链接: cuDNN Archive
重要!!如果你的模型要用到特定版本的pytorch,记得先看看跟这个版本的pytorch相匹配的cudnn: Previous PyTorch Versions
找到跟此前安装的cuda版本兼容的版本,点击进入选择对应服务器系统类型的安装包即可
下载完成后上传到服务器,找到合适的路径解压缩安装包:
然后将相应文件复制到cuda的安装路径,并修改权限即可,其中cudnn-linux-x86_64-8.9.4.25_cuda11-archive/为解压缩后的文件路径,/usr/local/cuda/为cuda的安装路径,按照自己的实际情况替换即可:
运行以下命令
若上述命令没有结果,则运行以下命令
若出现以下信息,恭喜你,cudnn安装成功!
准备就绪,我们可以安装anaconda,新建一个虚拟环境进行开发啦~~
往期传送门: 李大锤:Ubuntu安装anaconda保姆级教程记录
参考:
【1】 ubuntu系统安装CUDA和CUDNN(CUDA安装)_哔哩哔哩_bilibili
【2】 Ubuntu18.04安装CUDA+cuDNN+TensorFlow+TensorRT 萤火