ubuntu22.04安装nvidia驱动、cuda以及pytorch记录,亲测可行

本文详细介绍了在Ubuntu22.04上安装Nvidia显卡驱动、CUDAToolkit以及PyTorch的步骤,包括解决安装过程中可能出现的问题,如驱动与CUDA版本匹配和避免系统错误。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >


前言

ubuntu22.04安装nvidia驱动、cuda以及pytorch记录,主要是为了纠正网上很多文章的错误,避免走弯路。
我是在笔记本电脑测试的,安装了win11和ubuntu22.04双系统。
笔记本配置是12代i7、16G内存、3060 6G 显卡。安装ubuntu的时候就会出现黑屏无法启动的问题,这是显卡驱动的缘故,在grub提示下按e,然后添加nomodeset就可以进入系统了,这个按网上的文章叙述操作,基本都是可以的。
此外,装好系统之后还需要装个anaconda或者miniconda,这个按网上说的操作就行了,我装的miniconda。


一、安装nvidia显卡驱动

这个有很多方法,我是采用比较懒的方式,直接打开附加驱动,然后选择一个nvidia专用驱动安装,我选的是闭源的,目前版本是535和545,我安装的是545。装好之后重启即可。
请添加图片描述

二、安装cuda-toolkit

装好nvdia显卡驱动后,打开终端运行

nvidia-smi

可以看到如下信息:
请添加图片描述

### Ubuntu 22.04安装 PyTorchCUDA 的兼容性配置 #### 准备工作 为了确保系统的稳定性和软件之间的兼容性,在开始之前建议更新系统并重启计算机。 ```bash sudo apt update && sudo apt upgrade -y reboot ``` #### 安装 NVIDIA 驱动程序 通过终端中的 `nvidia-smi` 命令确认当前使用的 GPU 驱动版本,并据此选择合适的驱动版本进行安装。对于30系列及以上GPU,通常需要支持 CUDA 11.x 或更高版本的驱动[^1]。 #### 安装 CUDA 工具包 前往[NVIDIA官方CUDA下载页面](https://developer.nvidia.com/cuda-downloads),根据自己的需求挑选适合的操作系统版本(此处为Ubuntu 22.04)、架构(x86_64)以及其他参数来获取相应的安装文件。下载完成后解压并执行安装脚本: ```bash cd /path/to/downloaded/file/ sudo sh cuda_<version>_linux.run ``` 完成上述操作之后记得设置环境变量以便后续能够正常使用CUDA工具链: ```bash echo 'export PATH=/usr/local/cuda/bin:$PATH' >> ~/.bashrc source ~/.bashrc ``` #### 编译 NCCL 库 (可选) 如果计划利用多GPU加速,则可能还需要单独编译NCCL库以获得更好的性能表现。可以从GitHub克隆仓库并按照说明文档逐步构建二进制包[^2]: ```bash git clone --recursive https://github.com/NVIDIA/nccl.git cd nccl/ make -j src.build sudo dpkg -i build/pkg/deb/libnccl*.deb ``` #### 安装 PyTorch 最后一步就是安装PyTorch框架本身了。考虑到不同版本间的依赖关系复杂度较高,推荐直接采用预编译好的whl格式轮子来进行快速部署。可以访问[PyTorch官方网站](https://pytorch.org/get-started/locally/)查询最新的安装指南,这里给出一个基于pip的方式作为例子: ```bash pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117 ``` 注意这里的cu117代表所选用的具体CUDA版本号,请根据自己实际情况调整该数值。 #### 试验证 一切就绪后可以通过简单的Python代码片段试整个流程是否成功: ```python import torch print(torch.cuda.is_available()) print(torch.__version__) ``` 如果输出显示True并且打印出了正确的PyTorch版本信息即表示安装无误。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值