Ubuntu 20.04安装CUDA 11.0、cuDNN 8.0.5

不知道咋弄的ubuntu20.04电脑的cuda驱动丢了,无奈需装PyTorch环境,只有CUDA11.0以上版本才支持Ubuntu20.04,所以安装了CUDA11.0、cuDNN8.0.5 为防止频繁在浏览器检索对应的贴子,今天记录一下。

一. 驱动安装

为防止驱动安装后没办法进入登陆界面,这没有必要进行单独下载安装,只要在操作界面进行点击安装就行。具体如下:在软件更新->附加驱动点击应用就行

  1. 驱动安装
    在这里插入图片描述
    2.驱动测试
sudo reboot
nvidia-smi

就可以看到这些信息,如果命令可用说明驱动安装完成。
在这里插入图片描述

二. cuda安装

首先在Nvidia官网下载适合自己机器的CUDA版本,官网下载,本次使用 runfile 的方式安装(用deb格式安装的话,会在安装过程中替换掉已经安装好的显卡驱动),如图所

### 安装准备 为了确保顺利安装CUDAcuDNN,在Ubuntu 20.04上操作之前,需先完成NVIDIA显卡驱动的安装[^2]。这一步骤至关重要,因为缺少合适的图形驱动可能导致后续软件无法正常工作。 ### 安装CUDA Toolkit 对于Ubuntu 20.04而言,推荐使用的CUDA版本为11.6或更高版本。具体到当前环境,建议采用CUDA 12.2作为目标版本进行安装[^1]。以下是详细的安装过程: #### 下载CUDA Installer 访问[NVIDIA官方下载页面](https://developer.nvidia.com/cuda-downloads),选择对应的操作系统(Linux)、发行版(Ubuntu)、版本(20.04),以及架构(x86_64)。按照提示获取适用于系统的安装文件。 #### 执行安装脚本 通过命令行执行所下载的`.run`文件来启动安装向导: ```bash sudo sh cuda_<version>_linux.run ``` 注意:在运行此命令前,请关闭任何可能干扰安装进程的服务或应用程序,特别是与GPU相关的服务。 ### 配置环境变量 成功安装后,编辑用户的shell配置文件(如`.bashrc`)以添加必要的路径设置: ```bash export PATH=/usr/local/cuda-<version>/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-<version>/lib64\ ${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 记得替换上述代码中的 `<version>` 占位符为实际安装CUDA版本号,并使更改生效: ```bash source ~/.bashrc ``` ### cuDNN库集成 针对已选定的CUDA版本,应匹配相应的cuDNN版本。例如,当使用CUDA 12.2时,应当配套安装cuDNN 8.x系列的一个稳定版本。可以通过以下方式获得并应用cuDNN包: 前往[NVIDIA cuDNN官网](https://developer.nvidia.com/rdp/cudnn-archive)注册账号并登录,找到适合于CUDA 12.2的cuDNN版本,通常会提供.tar.gz压缩包形式供下载。 解压后将其中的内容复制至相应目录下: ```bash tar -xzvf cudnn-linux-x86_64-*.*.*_*-cuda*.tgz sudo cp cuda/include/* /usr/local/cuda/include/ sudo cp cuda/lib64/* /usr/local/cuda/lib64/ ``` 最后再次更新动态链接器缓存: ```bash sudo ldconfig ``` 至此,整个安装流程结束。可以尝试编译简单的CUDA程序验证安装是否成功。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Starry-sky(jing)

一起共勉,一起加油!!!

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值