找到对应版本的显卡驱动、cuda和pytorch安装的版本

本文记录了在单位电脑上安装深度学习环境的过程,遇到的问题及解决办法。使用的是NVIDIA QUADRO 2000显卡,尝试安装最新版本的CUDA(11.7)和PyTorch,但发现PyTorch不支持CUDA 11.7。通过官方高级驱动搜索找到旧版驱动472.47,该版本对应CUDA 11.4。下载并安装了CUDA 11.4,成功解决了版本兼容问题。

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

今天在单位电脑安装深度学习的环境

使用的NVIDIA QUADRO P2000的显卡,点进nvidia官网下载显卡驱动和cuda都是最新版本(目前最新版本11.7),但pytroch安装并没有11.7版本

 在官网(https://www.nvidia.cn/Download)找了一圈都没找到以前版本的显卡驱动和cuda下载

最后在网上找到了

以往显卡驱动下载地址(官方高级驱动搜索 | NVIDIA)找到对应的显卡下载即可

但我并不知道怎么下载下来的显卡驱动满足条件的cuda,所以我是一个个试的

最后选择的显卡驱动版本为472.47,对应的为cuda11.4

再下载旧版的cuda版本(CUDA Toolkit Archive | NVIDIA Developer),下载cuda版本

### Ubuntu 下 CUDAPyTorch版本兼容性及安装教程 #### 兼容性概述 在构建深度学习环境时,确保操作系统、CUDA PyTorch版本相互兼容是非常重要的。Ubuntu 20.04 被广泛认为是一个稳定的选择,因为它提供了较低版本显卡驱动程序,这使得它更容易与特定版本CUDA PyTorch 进行匹配[^1]。 以下是常见的 Ubuntu、CUDA PyTorch 版本组合: | **Ubuntu** | **CUDA** | **PyTorch** | |------------|-------------|--------------| | 20.04 | 11.3 | 1.10 | | 20.04 | 11.7 | 1.12 | | 22.04 | 11.8 | 1.13 | 上述表格中的版本经过测试能够较好地协同工作。 --- #### 安装步骤 ##### 1. 更新系统并安装依赖项 在开始之前,请先更新您的 Ubuntu 系统,并安装必要的开发工具库文件: ```bash sudo apt update && sudo apt upgrade -y sudo apt install -y build-essential cmake git unzip pkg-config libopenblas-dev liblapack-dev python3-numpy python3-scipy python3-matplotlib python3-yaml ``` 此操作将为后续的软件包安装提供基础支持[^2]。 ##### 2. 安装 NVIDIA 驱动程序 为了使 GPU 加速生效,您需要安装适合硬件的 NVIDIA 显卡驱动程序。可以通过以下命令自动检测并安装最新的驱动程序: ```bash sudo ubuntu-drivers autoinstall ``` 完成后重启计算机以加载新的驱动程序。 验证驱动程序是否正确安装: ```bash nvidia-smi ``` 如果显示了 GPU 使用情况,则说明驱动已成功安装[^3]。 ##### 3. 安装 CUDA 工具包 根据目标 PyTorch 版本选择对应CUDA 版本。例如,如果您计划使用 PyTorch 1.10,则应安装 CUDA 11.3。 下载地址可以从官方 CUDA 页面获取。或者通过 APT 包管理器完成安装: ```bash wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.0-1_all.deb sudo dpkg -i cuda-keyring_1.0-1_all.deb sudo apt-get update sudo apt-get -y install cuda-11-3 ``` 设置环境变量以便于访问 CUDA 命令: ```bash echo 'export PATH=/usr/local/cuda-11.3/bin:$PATH' >> ~/.bashrc echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc source ~/.bashrc ``` 验证 CUDA 是否正常运行: ```bash nvcc --version ``` ##### 4. 安装 cuDNN (可选) cuDNN 是一种用于优化深度神经网络性能的库。通常情况下,PyTorch 自带的功能已经足够满足大多数需求;但如果需要更高的效率,可以选择手动安装 cuDNN。 从 NVIDIA 开发者网站下载适用于所选 CUDA 版本的 cuDNN 文件,并按照文档指示解压到 `/usr/local/cuda` 中。 ##### 5. 安装 PyTorch 前往 [PyTorch 官方页面](https://pytorch.org/get-started/previous-versions/) 查找历史版本信息。假设我们需要安装 PyTorch 1.10 并搭配 Python 3.x CUDA 11.3,执行如下 pip 命令即可: ```bash pip install torch==1.10.0+cu113 torchvision==0.11.1+cu113 torchaudio===0.10.0 -f https://download.pytorch.org/whl/torch_stable.html ``` 注意替换 `torch`, `torchvision`, 其他组件的具体版本号来适应不同的配置需求。 ##### 6. 测试 GPU 支持状态 最后一步是确认 GPU 功能已被激活。打开 Python 解释器并输入以下脚本来检查设备可用性具体型号: ```python import torch print(f"CUDA Available: {torch.cuda.is_available()}") print(f"Device Count: {torch.cuda.device_count()}") print(f"Current Device Name: {torch.cuda.get_device_name(0)}") ``` 当输出表明存在至少一块活动的 GPU 设备时,即代表整个流程顺利完成。 --- ###
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

打赏作者

zlbbme_

你的鼓励将是我创作的最大动力

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

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

打赏作者

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

抵扣说明:

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

余额充值