使用Conda安装PyTorch出现CUDA版本不符问题求助

问题根源 #

你指定安装pytorch-cuda=11.8但实际得到CUDA 12.1，大概率是conda通道优先级或依赖冲突导致自动升级了CUDA runtime；同时你的NVIDIA驱动版本为11.2，不支持CUDA 12.1（CUDA runtime要求驱动版本不低于其最低要求，CUDA 12.1对应驱动需≥530.30.02，而你的11.2版本远低于此标准）。

修复步骤 #

1. 彻底清理现有PyTorch相关包 #

先卸载已安装的PyTorch组件，避免残留版本干扰：

conda uninstall pytorch torchvision torchaudio pytorch-cuda -y
conda clean --all -y

2. 安装匹配驱动版本的PyTorch #

你的驱动版本11.2最高支持CUDA 11.2 runtime，因此需要安装基于CUDA 11.2编译的PyTorch版本。使用以下conda命令（指定明确版本号确保匹配）：

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.2 -c pytorch -c nvidia

注：PyTorch 2.0.1是稳定支持CUDA 11.2的版本，若需其他版本可确认官方文档的CUDA支持情况，但务必保证CUDA runtime版本≤驱动支持的最高版本。

3. 验证安装结果 #

激活conda环境后，运行Python验证：

import torch
print(torch.version.cuda)  # 预期输出: '11.2'
print(torch.cuda.is_available())  # 预期输出: True，无警告

4. 额外排查建议 #

• 如果仍有问题，创建全新conda环境隔离依赖：

  conda create -n torch_env python=3.8 -y
  conda activate torch_env
  # 再执行步骤2的安装命令
  

• 避免混合使用pip和conda安装PyTorch，两种包管理工具的依赖处理逻辑不同，易导致版本混乱。

内容的提问来源于stack exchange，提问作者A. Sun