当前位置:首页 > 系统教程 > 正文

50系显卡Ubuntu 22.04深度学习环境搭建(RTX 5090/5080驱动+CUDA+cuDNN+PyTorch一站式教程)

随着2025年NVIDIA 50系显卡的全面普及,不少开发者在Ubuntu 22.04系统下遇到了环境配置难题。本文将为您提供一份详尽的50系显卡驱动安装及深度学习环境配置指南,即使是小白也能轻松上手。

SEO关键词:50系显卡驱动安装、Ubuntu CUDA配置、RTX 5090开发环境、PyTorch GPU安装

一、前期准备与系统更新

在开始之前,请确保您的主板已关闭安全启动(Secure Boot),否则驱动可能无法加载。

sudo apt-get updatesudo apt-get upgrade

二、安装NVIDIA 50系专用驱动

对于最新的50系显卡(如RTX 5090),建议使用官方PPA源安装。执行以下命令查看推荐驱动:

ubuntu-drivers devices

通常选择版本号最高的推荐驱动(例如 nvidia-driver-570 或更高):

sudo apt install nvidia-driver-570

安装完成后重启电脑,输入 nvidia-smi 检查是否成功。

50系显卡Ubuntu 22.04深度学习环境搭建(RTX 5090/5080驱动+CUDA+cuDNN+PyTorch一站式教程) 50系显卡驱动安装  Ubuntu CUDA配置 RTX 5090开发环境 PyTorch GPU安装 第1张

三、CUDA Toolkit 安装(2025年7月版)

50系显卡建议安装 Ubuntu CUDA配置 环境下的 CUDA 12.x 系列。前往NVIDIA官网下载对应的runfile文件:

wget https://developer.download.nvidia.com/compute/cuda/12.x.x/local_installers/cuda_12.x.x_linux.runsudo sh cuda_12.x.x_linux.run

注意:在安装界面中,按空格键取消勾选“Driver”(因为我们已经装好了),然后选择Install。最后在~/.bashrc中添加环境变量。

四、cuDNN 加速库配置

下载与CUDA版本匹配的cuDNN压缩包,解压并拷贝至CUDA目录:

sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/includesudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*

五、Anaconda 与 PyTorch 环境配置

为了在 RTX 5090开发环境 中保持整洁,我们使用Anaconda管理环境:

  • 创建环境:conda create -n torch_env python=3.10
  • 激活环境:conda activate torch_env
  • 安装 PyTorch GPU安装 版本:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu12x

六、验证测试

进入Python,输入以下代码测试:

import torchprint(torch.cuda.is_available())print(torch.cuda.get_device_name(0))

如果输出 True 和你的显卡型号,恭喜你,2025年最强开发环境已搭建完成!