最近有个框架要安装环境,环境有一定的版本要求,装的过程中发现安装的环境无法编译成功,多方查找原因后发现原来是CUDA版本太低了,于是升级了一下cuda版本,过程中也发现自己在Ubuntu系统上装CUDA比较生疏,于是大致记一下过程,为以后做参考。
(由于我是在windows下连的Linux服务器上装的,部分界面无法截图拷出,但是执行的命令基本不差,最后也是安装上了,并且编译过了,通过了验证)

首先是出现的错误,编译过程的错误是缺少依赖来着–dependency–xx,具体忘了
在有问题的环境下运行测试代码错误如下:

error in modulated_deformable_im2col_cuda: invalid device function

在这里插入图片描述
然后是基于问题去解决安装新的CUDA,
CUDA资源的官网(cuda toolkit):https://developer.nvidia.com/cuda-downloads
是深度学习框架,还要装cuDNN:https://developer.nvidia.com/rdp/cudnn-download
具体下什么版本就看需求了
我装的版本是cuda11.1.0,装的分别是.run文件和cuda11.x下对应的.xz格式的cudnn,其他格式文件的安装方法目前还不熟悉,总之效果应该是一样的
在这里插入图片描述
在这里插入图片描述
一、接下来就是安装步骤了,先安装cuda,就是这个文件 cuda_11.1.0_455.23.05_linux.run
在命令窗口进入放该文件的路径下,执行命令:

sudo sh cuda_11.1.0_455.23.05_linux.run

执行后会有弹窗,在驱动driver上按空格,再按下,到install,安装,输入accept,执行结果如下
在这里插入图片描述
正常的话,基本上cuda就安装好了。
可发现cuda安装在 /usr/local/cuda11.1/ 路径下,nvcc -V可查看cuda版本信息,若没有,可用下面命令添加,也可不用执行,供参考
在这里插入图片描述

sudo vim /etc/profile
source /etc/profile

在这里插入图片描述
下一张是个人界面,供个人参考
在这里插入图片描述
二,装cudnn,进入文件夹下,解压

tar xvf cudnn-linux-x86_64-8.4.0.27_cuda11.6-archive.tar.xz

在这里插入图片描述
然后是把解压后的部分文件拷到cuda11.1下面,其实和在windows上一样,进入解压后的cudnn文件夹,执行拷贝命令:

sudo cp -r include/* /usr/local/cuda-11.1/targets/x86_64-linux/include/
sudo cp -r lib/* /usr/local/cuda-11.1/targets/x86_64-linux/lib/

在这里插入图片描述
到此就安装结束了

Logo

更多推荐