最近有个框架要安装环境,环境有一定的版本要求,装的过程中发现安装的环境无法编译成功,多方查找原因后发现原来是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/
到此就安装结束了
|