提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
文章目录
一、卸载cuda和cudnn
cd /usr/local/cuda/bin
sudo ./cuda-uninstaller
sudo apt-get remove cuda
sudo apt autoremove
sudo apt-get remove cuda*
cd /usr/local
sudo rm -r cuda-版本号 #删除文件
二、安装cuda
cuda官网下载地址,—》下载
会很卡,找自己想要下载的版本,我是10.2
wget https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_440.33.01_linux.run
自动下载中。。。。。
#cd 到Downloads目下
sudo sh cuda_10.2.89_440.33.01_linux.run
进入安装界面,点Continue
输入accept,回车
键盘方向键选择Driver,按enter将driver选项去掉,这是让你装驱动,不要装。然后install
安装结束,可以看到驱动没有装上,然后其他两个都装上了,可以了
1.环境变量配置
gedit ~/.bashrc
在弹出的窗口最下面输入环境变量,复制粘帖过去就行
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
然后save 后在终端source 下
source ~/.bashrc
2.测试版本号
测试cuda 有没有装上,可以查到版本号,阔以了
nvcc -V
三、安装cudnn(推荐deb 安装)
官方下载地址下载
勾选agree,选择要下载的版本
选择deb安装方案,下载对应的三个包
cd到你deb包的具体位置(我包是自己建在Downloads 下)
就是sudo dpkg -i 文件名这样的格式
$ sudo dpkg -i libcudnn8_8.2.4.15-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn8-dev_8.2.4.15-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn8-samples_8.2.4.15-1+cuda10.2_amd64.deb
1.测试是否安装上
cd /usr/src/cudnn_samples_v8
cp -r /usr/src/cudnn_samples_v8/ $HOME
cd $HOME/cudnn_samples_v8/mnistCUDNN
sudo make clean
sudo make
./mnistCUDNN
V8还是V7输入是看你自己cudnn版本
2.执行“sudo make”时候可能错误,出现fatal error ,FreeImage.h
输入
sudo apt-get install libfreeimage3 libfreeimage-dev
重新sudo make 就可以了
3.测试结果
如果是test passed 则安装成功。
4.测试结果(补充方案二)
如果上面测试没法成功,还有方案二测试方案
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
老版用上一种,新一点用下一种
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
5.测试结果(补充方案三)
当都找不到文件的的话
可能是cudnn_version.h的地址不对
whereis cudnn_version.h
把下面/usr/linclude/地址换成你找到的地址(成功!)
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2