ubuntu 安装 cuda cudnn

本文档详细介绍了如何下载并安装CUDA和cuDNN。首先,确认显卡驱动兼容的CUDA版本,然后从NVIDIA官网下载合适的CUDA安装包。在安装过程中,务必取消自带驱动选项。接着,下载与CUDA版本匹配的cuDNN,将其头文件和库文件复制到CUDA安装目录。最后,更新环境变量,使CUDA和cuDNN生效。通过nvcc-V和nvidia-smi命令检查安装是否成功。
部署运行你感兴趣的模型镜像

下载 cuda和 cudnn

  1. 了解自己机器的nvidia 显卡驱动支持什么版本的cuda 最新的信息参照官网的文档, 下图是我在写这篇博客的最新配置,第一列是cuda 版本,第二列是 linux 驱动版本, 第三列是windows 驱动版本。
    在这里插入图片描述

  2. 官网下载 你的机器需要的cuda 版本,下图是官网截图的一部分
    在这里插入图片描述

  3. 根据自己的操作系统类型,操作系统版本去选择合适的安装包,这里我的是linux->ubuntu->18.04->runfile
    在这里插入图片描述

  4. 长时间等待cuda的下载, 下载成功后会有一个cuda*.run 的文件,这个就是我们需要安装的文件了

  5. 选择需要的cudnn版本,同样的,cudnn 要和nvidia 显卡驱动以及cuda 版本要兼容,具体的信息看官网链接
    下图是官网截图的最新版本的support matrix,进入官网可以查看全部。
    在这里插入图片描述

  6. 下载对应的cudnn 版本,官网链接 需要登录!!!!

安装cuda

  1. 保证安装的用户有root 权限
  2. bash cuda***.run
  3. 根据提示安装就可以了,这里注意一点,cuda 本身自带显卡驱动,在选择的时候一定要把这一项取消,否则它会覆盖掉原来的显卡驱动
  4. 安装完成,系统默认安装在了/usr/local 里,你会发现这里面多了 cudacudaX.X 文件夹,后者是真正的安装结果,cuda 只是链接它的。
  5. 配置环境变量:
    vim /etc/bash.bashrc
    最后一行加入export PATH=${PATH}:/usr/local/cuda/bin
    再加入一行export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/cuda/lib64
    source /etc/bash.bashrc 更新环境变量
    6.到这里,cuda就完成安装了

安装 cudnn

  1. 解压刚才下载的cudnn 压缩包
    tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz 会生成一个cuda 文件夹
  2. 复制文件
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include  # root 用户去掉sudo
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64    
  1. 更改权限
    $ sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

安装完成

nvcc -V 查看是否成功

版本不一样?

nvidia-smi 显示的cuda 版本和nvcc -V显示的cuda的版本不一样,不用担心,跑程序的时候用的是nvcc -V 对应的cuda

您可能感兴趣的与本文相关的镜像

PyTorch 2.9

PyTorch 2.9

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值