当前位置:   article > 正文

Ubuntu 安装 CUDA 和 cuDNN 详细步骤_ubuntu安装cuda和cudnn

ubuntu安装cuda和cudnn

我的Linux系统背景:

  • 系统和驱动都已安装。

  • 系统是centos 8。查看自己操作系统的版本信息:cat /etc/issue或者是 cat /etc/lsb-release

  • nvidia-smi可以看到显卡驱动和可支持的最高cuda版本,我的是12.2。驱动版本是535.129.03

首先,检查是否安装了CUDA:ls -l /usr/local | grep cuda

http:// https://developer.nvidia.com/cuda-toolkit-archive 如下截图, 选择自己系统对应的 CUDA 版本,

若第1步提示Existing package manager installation of the driver found. It is strongly recommended that you remove this before continuing.,选择continue,继续操作

  1. Do you accept the above EULA? (accept/decline/quit):
  2. │ accept #输入accept
  3. │─────────────────────────────────────────────────────
  4. #安装选项,由于我已经安装有Driver: 535.54.03,所以没有选择。
  5. │ CUDA Installer
  6. │ - [ ] Driver
  7. │ [ ] 535.54.03
  8. │ + [X] CUDA Toolkit 10.2
  9. │ [ ] CUDA Samples 10.2
  10. │ [ ] CUDA Demo Suite 10.2
  11. │ [ ] CUDA Documentation 10.2
  12. │ Options
  13. │ Install #[ ]不选择,带X的是需要安装的部分,之后选择 Install
  14. ————————————————

安装好后显示 

输入 ll /usr/local/

可以看到 /usr/local/ 目录下已经有 cuda -> /usr/local/cuda-10.2// 软链接。

添加环境变量

打开 ~/.bashrc

  1. export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
  2. export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

验证安装成功

最后输入命令 nvcc -V 有如下输出,说明已安装成功。

最后这步有问题在打开 ~/.bashrc中重写下面三行

  1. export PATH=$PATH:/usr/local/cuda/bin
  2. export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
  3. export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda/lib64

nvcc -V

cuDNN的安装与检测

3.1. cuDNN的安装

NVIDIA官网的cudnn下载页面上下载与安装CUDA对应的cudnn,网址为Log in | NVIDIA Developer。选择Ubuntu2

cudnn-linux-x86_64-8.8.0.121_cuda12-archive.tar.xz

  1. sudo cp cuda/include/cudnn.h /usr/local/cuda/include
  2. sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
  3. sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

参考:https://blog.csdn.net/wohu1104/article/details/107041387

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/2023面试高手/article/detail/627465
推荐阅读
相关标签
  

闽ICP备14008679号