Ubuntu20.4+CUDA11.1+ CUDNN8.05

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

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值