AssertionError: Torch not compiled with CUDA enabled cuda12.6
时间: 2025-04-27 11:29:32 浏览: 25
### 解决 Torch 未编译启用 CUDA 的方法
当遇到 `AssertionError: Torch not compiled with CUDA enabled` 错误时,这表明当前安装的 PyTorch 版本不支持 CUDA 或者环境配置存在问题。为了确保能够正常使用 GPU 加速功能,建议按照以下方案处理:
#### 验证现有安装情况
首先确认已有的 PyTorch 安装是否确实带有 CUDA 支持。可以通过 Python 控制台执行如下命令来检测:
```python
import torch
print(torch.cuda.is_available())
```
如果返回 False,则说明当前环境中使用的 PyTorch 不具备所需的硬件加速能力。
#### 卸载旧版本 PyTorch
为了避免冲突并确保新版本正确安装,在重新安装之前先移除现有的 PyTorch 库:
```bash
pip uninstall torch torchvision torchaudio
```
#### 获取适用于特定 CUDA 版本的新版 PyTorch
对于希望利用 CUDA 12.6 进行计算的情况,应该下载由官方预构建且兼容该版本显卡驱动程序的二进制文件。访问 [PyTorch官方网站](https://pytorch.org/get-started/locally/) 并根据操作系统和个人偏好选择合适的安装方式与参数组合[^1]。
例如针对 Windows 用户通过 pip 工具获取最新稳定版(假设已经安装了对应版本的 CUDA Toolkit 和 cuDNN),可以运行下面这条指令完成安装过程:
```bash
pip install --upgrade torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu126
```
成功完成后再次测试 cuda 是否可用即可验证问题是否得到解决。
阅读全文
相关推荐











