AssertionError: Torch not compiled with CUDA enabled:先确认装入的是 CPU 还是 CUDA 构建

简介: 本文详解PyTorch“AssertionError: Torch not compiled with CUDA enabled”错误的根源与精准排查方法,厘清`nvcc`/`nvidia-smi`/`torch.version.cuda`三类CUDA版本差异,强调CPU包无法靠装CUDA工具包激活,并提供环境隔离、构建验证及安全回退的完整实践路径。(239字)

原帖使用 Windows 10、Python 3.7 和 PyTorch 1.2.0;包列表还包含 cpuonly。调用 CUDA 时出现:

AssertionError: Torch not compiled with CUDA enabled

分清三个不同的 CUDA 版本信息

系统里的 nvcc 对应开发工具链,nvidia-smi 显示驱动及其支持的 CUDA 上限,torch.version.cuda 则说明当前 PyTorch 包的构建信息。三者不必逐字相同。原帖同时列出了 cudatoolkit 与 cpuonly,恰好说明“装过 CUDA 工具包”不足以判断 torch 是哪种构建。

先查导入结果,再看安装记录,可以避免反复重装驱动却一直使用 CPU 包。

安装了系统 CUDA Toolkit,并不能把 CPU 版 torch 变成 CUDA 版。在发生 .cuda() 之前检查实际导入的包:

import sys
import torch
print("josh050608")
print("python:", sys.executable)
print("torch:", torch.__version__, torch.__file__)
print("build CUDA:", torch.version.cuda)
print("available:", torch.cuda.is_available())

若 torch.version.cuda is None,当前就是没有 CUDA 支持的构建。若它有版本号但 is_available() 为 False,则转查 NVIDIA 驱动、GPU 可见性与容器权限,不能再把原因定成 CPU 构建。

复现原帖旧环境时,官方历史页给出的 1.2.0/CUDA 10.0 Conda 组合如下。先建一个独立的 Python 3.7 环境,再在已激活的环境执行,避免既有 cpuonly 约束影响解析:

conda create -n torch120 python=3.7
conda activate torch120
conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

这只是历史复现组合。新项目应按官方选择器选择受支持的 Python、驱动和构建,不应把旧命令当作最新推荐。没有兼容 NVIDIA GPU 时,应保留 CPU 执行路径:device = torch.device('cuda' if torch.cuda.is_available() else 'cpu'),并使用 model.to(device) 和 inputs.to(device)。

最后重新启动 Python,重复检查,并用 torch.ones(1, device='cuda') 验证真正的 CUDA 分配。

检查结果对应哪条修复路线

build CUDA available 应当继续检查什么
None False 当前 torch 无 CUDA 支持;核对解释器后换适合平台的构建
有版本号 False 驱动、GPU 可见性、容器设备映射;此时不等于装了 CPU 包
有版本号 True 先做小分配,再回到仓库中检查硬编码的设备编号

在原来启动仓库的同一终端执行 python -m pip show torch;Conda 用户再执行 conda list pytorch、conda list cpuonly。这一步看的是包的来源和约束,不应混着执行多套历史安装命令。若选择新建环境,应先在新环境验证小程序成功,再让 IDE、Notebook 和仓库入口一起切过去。

下面是安装检查通过后的最小计算,不加载模型,不下载数据。它会明确拒绝静默退回 CPU,适用于“本来就要求使用 CUDA”的验证:

import torch
if torch.version.cuda is None:
    raise RuntimeError("当前导入的是无 CUDA 支持的 torch 构建")
if not torch.cuda.is_available():
    raise RuntimeError("构建支持 CUDA,但本进程尚不能使用 GPU")
x = torch.arange(4, dtype=torch.float32, device="cuda")
y = x.square()
torch.cuda.synchronize()
assert y.device.type == "cuda"
assert y.cpu().tolist() == [0.0, 1.0, 4.0, 9.0]
print("device:", torch.cuda.get_device_name(0), y.device)

若项目允许 CPU,则同时迁移模型、输入和目标,替换仓库中直接写死的 .cuda()。只修改一个 device 变量,不会自动改变其他文件中的 .cuda() 调用。历史环境的 1.2.0 组合适合复现旧项目;新 Python 不能直接照装旧 wheel,具体支持范围以目标版本的官方发行记录为准。

参考:原始求助帖;技术依据:PyTorch 历史安装组合、PyTorch 官方安装指南、PyTorch CUDA 语义(2.5)。

相关文章
|
11天前
|
Linux 开发工具 iOS开发
Codex保姆级教程(下载+安装+登录+使用,含所有版本,2026最新)
Codex是OpenAI推出的开源AI编程智能体,本地运行,可理解整个代码库,支持自然语言指令改代码、执行命令、跑测试、修Bug,兼容Win/macOS/Linux。(239字)
4161 22
|
11天前
|
JavaScript API iOS开发
DeepSeek Harness下载、安装、配置一篇搞定(2026整理,送常用插件)
DeepSeek Harness(DSH)是DeepSeek于2026年开源的MIT协议AI Agent框架,支持插件化组装模型、工具、沙箱与UI,兼容主流大模型。本文提供免Node.js的桌面端安装包(Win/macOS),含详细图文教程及热门插件推荐。
4211 49
|
6天前
|
存储 弹性计算 人工智能
2026 年阿里云双十一活动入口攻略,新客服务器优惠通道、三大会场入口梳理
综合来看,2026年阿里云双十一活动入口清晰: 阿里云轻量应用服务器、阿里云ECS云服务器、阿里云GPU云服务器、阿里云百炼大模型服务平台 活动会场、综合权益中心、99计划专场三大入口覆盖服务器、AI权益与续费特价;新用户通过控制台首页banner、权益中心、产品页三处路径进入专属专场,老用户通过控制台弹窗和费用中心权益中心进入老客专区,企业用户做企业实名后解锁专属满减与阶梯折扣。
821 0
|
12天前
|
iOS开发 Docker MacOS
【2026实测】Docker Desktop下载、安装、汉化、配置教程(附官网安装包)
Docker Desktop 是 Docker 官方推出的桌面容器管理工具,集成引擎、CLI、图形界面与 Kubernetes,支持 Win/macOS/Linux。2026 最新版 4.88.1 强化 ARM 原生运行、镜像加速及资源可视化,附中文汉化与国内镜像配置指南。
2025 7
|
6天前
|
云安全 人工智能 安全
|
5天前
|
Linux 开发工具 iOS开发
2026最新codex下载安装保姆级使用教程(下载+安装+登录+使用,含所有历史版本)
本文详解OpenAI Codex国内快速上手指南:提供Win/macOS/Linux全平台免安装包、npm/Homebrew等多方式安装、账号/API Key双登录、简体中文切换,覆盖下载、安装、汉化、使用及高频避坑(如微软商店失败、npm超时等),10分钟即可跑通。
1167 3
|
12天前
|
安全 算法 定位技术
《高性价比人生指南》完整原版:普通人用最少的资源,过好这一生的底层逻辑~How To Live Better
📖《高性价比人生指南》是一份基于实证、拒绝焦虑的生存手册,倡导以“人生时薪”为视角,重新计算时间、金钱与精力的成本与收益。它不承诺逆袭,只教人算清账:省下的钱若耗掉时间便亏;买体验比买物品更值;睡眠是最高杠杆的健康投资;关系贵在情绪价值而非表面热络;学习重在改变行为而非自我安慰。核心理念:用低成本、高回报的微小行动,换取长期的选择权——不是成为谁,而是掌控自己人生的主动权。
2150 6
|
云安全 安全 网络安全
|
3天前
|
人工智能 安全 API
DeepSeek Harness下载、安装、配置一篇搞定(2026整理,送常用插件)
DeepSeek Harness(dsh)是DeepSeek AI于2026年开源的轻量级Agent运行框架,基于Cordis内核,践行“一切皆插件”理念。支持Web/桌面双端,开箱即用;内置多模型接入、丰富插件生态(超3.5万),适配编程、办公、创作等场景。(239字)
770 0

热门文章

最新文章