AI 助理
文档备案控制台

PyTorch

首页 标签 PyTorch
# PyTorch #
关注
5135内容
最新 热门 文章 问答
nick3343899940-48023
|
1天前
|
博文
Expected all tensors to be on the same device, cuda:0 and cpu：恢复训练时检查 SGD 动量
该问题源于断点续训时优化器状态（如SGD动量缓冲区）未与模型参数同设备，导致`optimizer.step()`中跨设备张量运算失败。前向/反向可正常执行，但动量更新报错“cuda:0 and cpu”。需检查并统一`param`、`grad`及`optimizer.state`中所有Tensor的设备。
# 并行计算 # PyTorch # 算法框架/工具 # 异构计算
nick3343899940-48023
|
1天前
|
博文
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字）
# 并行计算 # PyTorch # 算法框架/工具 # 异构计算 # Python
nick3343899940-48023
|
1天前
|
博文
RuntimeError: CUDA error: device-side assert triggered：Colab 先重启，再找首次断言
该文详解CUDA断言错误调试策略：强调先重启运行时、启用`CUDA_LAUNCH_BLOCKING=1`，用最小张量验证环境；区分“会话已坏”与新操作触发，主张逐段定位而非臆测标签越界，提供实用边界检查函数与验证流程。（239字）
# 并行计算 # PyTorch # Linux # 算法框架/工具 # 索引
nick3343899940-48023
|
1天前
|
博文
No module named "torch"：Notebook 能导入，终端脚本却失败时先对齐解释器
该问题本质是Python环境不一致：Notebook与控制台使用了不同解释器。需比对`sys.executable`路径，确认torch是否安装在当前运行环境中，并用`python -m pip`精准安装，避免路径错配。（239字）
# 并行计算 # IDE # PyTorch # 开发工具 # Python
nick3343899940-48023
|
1天前
|
博文
ValueError: only one element tensors can be converted to Python scalars：张量列表先检查形状，再 stack
该文解析PyTorch中`list(tensor)`引发的`ValueError`根源：列表含多元素张量时，不可直接转Python标量。强调应通过`tensor.shape`或遍历打印各张量尺寸来查形状；同形张量用`torch.stack`（增维）而非`cat`或`item`；异形则保留原结构，避免强行转换。
# PyTorch # 算法框架/工具 # 索引 # Python
nick3343899940-48023
|
1天前
|
博文
No module named "torch"：Notebook 能导入，终端脚本却失败时先对齐解释器
该问题本质是Python环境不一致：Notebook与控制台使用了不同解释器，导致torch仅安装在其中一个环境中。关键在于比对`sys.executable`路径，确认实际运行环境，并在对应环境精准安装（推荐`python -m pip install torch`），而非盲目重装。
# 并行计算 # IDE # PyTorch # 开发工具 # Python
nick3343899940-48023
|
1天前
|
博文
Expected object of scalar type Float but got scalar type Double for argument #4 'mat1'：NumPy 输入与 Linear 权重怎样统一类型
PyTorch中`RuntimeError: Expected Float but got Double`常因输入Tensor（float64）与模型权重（float32）类型不一致导致。关键在第一层Linear的输入与权重dtype需统一：可将数据转为float32（推荐），或模型整体升为float64。务必检查`tensor.dtype`而非数值显示，并确保`loss.backward()`前类型已对齐。（239字）
# PyTorch # 算法框架/工具 # 异构计算 # 索引 # Python
nick3343899940-48023
|
1天前
|
博文
PytorchStreamReader failed finding central directory：AlexNet 预训练权重先查缓存文件
该错误表明PyTorch加载预训练AlexNet权重时，本地缓存的`.pth`文件损坏（非完整ZIP格式），常见于下载中断或保存异常。需定位并验证`torch.hub`缓存中的权重文件，检查是否为HTML错误页或截断文件，再备份后重新下载，而非强行修改或跳过校验。（239字）
# 缓存 # PyTorch # 算法框架/工具 # 异构计算 # Python
nick3343899940-48023
|
1天前
|
博文
element 0 of tensors does not require grad：机器人动作后的观测损失不能自动反传
原帖将策略输出`detach()`转为NumPy驱动机器人，导致计算图断裂；新图像坐标无法反向传播至策略网络。错误`element 0 of tensors does not require grad...`正源于此——梯度在`detach()`处中断，非类型或设备问题。需保留张量路径或改用策略梯度/有监督训练。（239字）
# 算法 # 机器人 # PyTorch # 算法框架/工具 # Python
nick3343899940-48023
|
1天前
|
博文
Can't call numpy() on Variable that requires grad：检查 detach 返回值是否真的被使用
该帖探讨PyTorch张量转NumPy时仍报“requires grad”错误的深层原因：`detach().cpu().numpy()` 仅作用于当前变量，若绘图/日志等处**重复使用原张量**或隐式转换，错误仍会发生。需打印`requires_grad`与设备状态定位真实调用点，并确保所有下游操作均基于已detach+cpu的新对象。（239字）
# 存储 # PyTorch # 算法框架/工具 # Python
免费试用