AssertionError: Torch not compiled with CUDA 12.7
时间: 2025-01-06 22:37:43 浏览: 40
### 解决 Torch 未使用 CUDA 12.7 编译引发的 AssertionError
当遇到 `AssertionError: Torch not compiled with CUDA enabled` 的错误提示时,表明当前使用的 PyTorch 版本并未支持所期望的 CUDA 版本。对于特定于 CUDA 12.7 和 PyTorch 的配置问题,可以采取如下措施来解决问题。
#### 验证现有环境设置
确认现有的 Python 环境以及已安装的 PyTorch 是否确实为 CPU-only 版本而非 GPU 支持版[^3]。可以通过运行以下代码片段验证:
```python
import torch
print(torch.cuda.is_available())
```
如果返回 False,则说明当前环境中 PyTorch 并未启用 CUDA 功能。
#### 安装适配 CUDA 12.7 的 PyTorch 版本
由于官方发布的预构建二进制文件可能不会立即提供最新 CUDA 版本的支持,因此建议通过指定确切版本号的方式手动下载并安装适合 CUDA 12.7 的 PyTorch 轮子文件(whl)。例如,假设要安装与 Python 3.x 兼容且针对 Linux x86_64 构建的 PyTorch 1.y.z 版本,可执行如下 pip 命令:
```bash
pip install https://download.pytorch.org/whl/cu127/torch-1.y.z-cp3x-cp3xm-linux_x86_64.whl
```
请注意替换上述 URL 中的具体版本号以匹配实际需求,并确保选择正确的操作系统架构和 Python 版本标签[^4]。
#### 更新或重新创建 Conda 环境
有时 conda 可能会默认安装 CPU-only 的 PyTorch 包。为了避免这种情况,在创建新的 conda 环境时应明确指明所需的 CUDA 版本。比如:
```bash
conda create -n pytorch_cuda python=3.x
conda activate pytorch_cuda
conda install pytorch torchvision torchaudio cudatoolkit=12.7 -c pytorch
```
这将帮助建立一个带有适当 CUDA 工具链的新环境,从而避免潜在的依赖冲突问题。
阅读全文
相关推荐










