0


ubuntu安装cuda和cudnn

1.下载cuda

CUDA Toolkit Archive | NVIDIA DeveloperPrevious releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production drivers appropriate for your hardware configuration.https://developer.nvidia.com/cuda-toolkit-archive

进入上方链接,下载对应的cuda,类型选runtime

可以直接使用wget下载,wget命令

2.下载cudnn

cuDNN Archive | NVIDIA DeveloperExplore and download past releases from cuDNN GPU-accelerated primitive library for deep neural networks.https://developer.nvidia.com/rdp/cudnn-archive进入官网,找到和cuda相匹配的cudnn版本,进入之后,下载的cudnn library如下图所示。点击之后,会直接下载本机,之后可以自行上传服务器。下载下来的格式为tgz格式,一般比较大,1个多G。

3.安装cuda

sudo sh cuda_11.2.1_460.32.03_linux.run

之后要求输入密码,等待几分钟(大概),等待下面界面的弹出:

然后进入下一个界面,回车取消驱动,选择install

后面就按规矩来就好了,其中有一个询问是否符号链接,即/usr/local/cuda指向当前的cuda,如果需要,选择yes,否则no

可以使用下述命令查看新安装的cuda

ls /usr/local/
ls /usr/local/cuda-11.2
ls /usr/local/cuda #和上面那条命令展示出来的内容是一样的,说明建立了软链接

4.安装cudnn

使用下面的命令解压

tar -zxvf cudnn-11.2-linux-x64-v8.1.1.33.tgz

然后依次执行

sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.2/include
sudo cp cuda/lib64/libcudnn*  /usr/local/cuda-11.2/lib64/
sudo chmod a+r  /usr/local/cuda-11.2/include/cudnn.h

cudnn的安装就完成了。

5.配置环境变量

vi ~/.bashrc

然后,在最后面添加或者编辑相关的代码大致如下:

export PATH=$PATH:/usr/local/cuda-11.2/bin
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

之前已经有cuda,就将原来的cuda版本改成当前的版本即可

6.查看cuda和cudnn版本

重启服务器,就可以发现cuda版本为新版本了

使用 nvcc -V查看cudnn版本

使用 cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2查看cudnn版本

为8.1.1

标签: ubuntu

本文转载自: https://blog.csdn.net/weixin_46455141/article/details/125507029
版权归原作者 木之希 所有, 如有侵权,请联系我们删除。

“ubuntu安装cuda和cudnn”的评论:

还没有评论