PackagesNotFoundError: The following packages are missing from the target environment: - torchaudio - cudatoolkit - pytorch - torchvision

### 解决方案概述 在安装 PyTorch 及其相关组件（`torchvision`、`torchaudio` 和 `cudatoolkit`）时，版本不匹配问题通常是由于以下几个原因引起的：操作系统支持范围不足、CUDA 工具链版本与硬件架构不适配以及 Python 版本约束。以下是针对该问题的全面解决方案。 --- ### 1. 明确环境需求 首先，需明确以下几点以缩小排查范围： - 当前使用的 **Python 版本** 是否被目标 PyTorch 版本支持[^2]。 - 系统中的 **CUDA 驱动版本** 是否满足最低要求。 - GPU 的计算能力（Compute Capability, CC）。例如，RTX 3050 Laptop GPU 的 CC 为 `8.6`，需要 CUDA 11.x 或更高版本的支持[^1]。 --- ### 2. 清理旧版本依赖 为了避免潜在冲突，建议卸载所有先前安装的相关包： ```bash pip uninstall torch torchvision torchaudio cudatoolkit ``` 如果是通过 Conda 安装，则执行以下命令： ```bash conda remove pytorch torchvision torchaudio cudatoolkit ``` --- ### 3. 正确选择并安装兼容版本 根据官方文档和实际需求，推荐以下两种主流方式之一进行安装： #### 方法 A: 使用 Pip 安装预构建二进制文件 Pip 提供了简单快捷的方式获取最新稳定版 PyTorch 及其配套库。对于 CUDA 11.7 支持的情况，可运行如下命令： ```bash pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117 ``` 此命令会自动拉取与 CUDA 11.7 兼容的 PyTorch、`torchvision` 和 `torchaudio` 版本组合[^1]。 #### 方法 B: 使用 Conda 创建独立环境 Conda 更适合于复杂的多依赖场景，尤其是当系统中有其他 CUDA 应用共存时。创建一个新的虚拟环境，并指定所需的 CUDA 工具链版本： ```bash conda create -n myenv python=3.9 conda activate myenv conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch -c nvidia ``` 注意这里的 `-c nvidia` 参数确保从 NVIDIA 官方通道获取最新的 CUDA 工具链[^1]。 --- ### 4. 验证安装结果 完成安装后，可以通过以下代码片段验证各模块是否正常加载及其版本信息： ```python import torch import torchvision import torchaudio print(f"PyTorch Version: {torch.__version__}") print(f"TorchVision Version: {torchvision.__version__}") print(f"Torchaudio Version: {torchaudio.__version__}") print(f"CUDA Available: {torch.cuda.is_available()}") print(f"CUDA Version: {torch.version.cuda}") ``` 如果一切顺利，输出应当显示正确的版本号且表明 CUDA 已启用。 --- ### 5. 常见错误处理 #### 错误 1: `No matching distribution found for cudatoolkit` 这种提示往往意味着本地 pip 缓存过期或网络连接不稳定。尝试清理缓存后再重试： ```bash pip cache purge pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117 ``` #### 错误 2: `ModuleNotFoundError: No module named 'torch'` 可能是未激活正确的 conda 环境所致。重新激活环境并检查路径设置： ```bash conda activate myenv which python ``` #### 错误 3: `RuntimeError: Found no NVIDIA driver on your system` 说明当前系统缺少必要的 NVIDIA 驱动程序。前往 [NVIDIA Driver Downloads](https://www.nvidia.com/Download/index.aspx)，下载并安装适用于您 GPU 的最新驱动。 --- ###