Ubuntu20.4+CUDA11.1+ CUDNN8.05
文章目录
前言
提示:这里可以添加本文要记录的大概内容:
例如:随着人工智能的不断发展,机器学习这门技术也越来越重要,很多人都开启了学习机器学习,本文就介绍了机器学习的基础内容。
提示:以下是本篇文章正文内容,下面案例可供参考
一、CUDA11.1安装?
1.CUDA11.1下载安装
CUDA:https://developer.nvidia.com/cuda-toolkit
根据系统配置选择相应版本,这里选择runfile(local)安装:
根据官方给出的命令输入
代码如下(示例):
wget https://developer.download.nvidia.com/compute/cuda/11.2.2/local_installers/cuda_11.2.2_460.32.03_linux.run
sudo sh cuda_11.2.2_460.32.03_linux.run
结果展示 :
安装位置: /usr/local/cuda-11.1(上图提示)
查看版本信息:nvcc -V
2.过程配置
选择runfile版本包含驱动文件,之前安装过可根据提示取消驱动安装的勾选。按空格即可取消。然后一路yes。
添加环境变量
vim .bashrc
export PATH="/usr/local/cuda-11.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-11.1/lib64:$LD_LIBRARY_PATH"
Source .bashrc
注意:环境变量中路径的cuda对应版本,这几个变量中的’cuda-11.1’可替换为’cuda’,然后在/usr/local/目录下给软链接cuda赋值,这样就方便安装多个cuda.
3.拓展_多CUDA配置
修改环境变量
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
删除之前的软连接,添加需要cuda版本的软连接
sudo rm -rf cuda
sudo ln -s /usr/local/cuda-11.1 /usr/local/cuda
二、CUDNN8.05安装
1.CUDNN8.05下载安装
cuDNN:https://developer.nvidia.com/cudnn
注:需要登入nvidia账号后才能下载
复制连接使用命令下载:
Weget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.5/11.1_20201106/cudnn-11.1-linux-x64-v8.0.5.39.tgz
进行解压:
tar zxvf cudnn-11.1-linux-x64-v8.0.5.39.tgz
解压后在当前目录会得到一个cuda文件夹,在当前目录输入以下命令:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
2.查看结果
在用cmak-gui安装openpose时,提示错误信息找不到cudnn,即判断cudnn没有配置成功。经过多次尝试,在解压后的cuda文件夹含有include和lib64两个文件。include中的cudnn.h包含很多引导信息(某篇说的,没有记录下来),还存在cudnn_version.h. 所以可能是没有将include中的文件完整导入/usr/local/cuda/include/。
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
没有任何显示
cat /home/wanhf/Downloads/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
显示版本信息。
所以替换这一命令:
sudo cp cuda/include/*.h /usr/local/cuda/include/
成功解决该问题,因为被搞的裂开了,尝试了很多方法,就没记录下来
总结
以上就是今天要讲的内容,本文仅仅简单介绍了cuda cudnn的安装过程以及碰见的一些问题,为openpose的环境搭建打下基础
参考文档:
https://blog.csdn.net/Thanlon/article/details/108279649
https://zhuanlan.zhihu.com/p/174654752