Pytorch错误：AssertionError: If capturable=False, state_steps should not be CUDA tensors

最新推荐文章于 2023-10-21 15:32:52 发布
最新推荐文章于 2023-10-21 15:32:52 发布
阅读量807 收藏
点赞数 1
分类专栏： 机器学习和深度学习之旅 文章标签： pytorch 人工智能 python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_42111770/article/details/133701408
版权

基础配置的版本信息如下所示：

版本号
pytorch1.12.0
cuda11.3
python3.10

现象描述：训练中断，从中断处再起训练时，爆出AssertionError: If capturable=False, state_steps should not be CUDA tensors的错误。

出错原因

解决方案：在stackOverflow中找到的答案。
在这里插入图片描述
不想降低训练速度，我将pytorch卸载，安装1.11版本的pytorch，解决了问题。

参考

  1. AssertionError: If capturable=False, state_steps should not be CUDA tensors
马鹤宁
关注
  • 1
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
专栏目录
解决pytorch报错:AssertionError: Invalid device id的问题
01-21
在服务器上训练的网络放到本地台式机进行infer，结果出现报错： AssertionError: Invalid device id ...以上这篇解决pytorch报错:AssertionError: Invalid device id的问题就是小编分享给大家的全部内容了，
If capturable=False, state_steps should not be CUDA tensors
QT-Smile
08-04 3259
If capturable=False, state_steps should not be CUDA tensors
If capturable=False, state_steps should not be CUDA tensors.
qq_30249871的博客
10-20 1642
If capturable=False, state_steps should not be CUDA tensors.
Pytorch 报错 AssertionError: If capturable=False, state_steps should not be CUDA tensors.
weixin_44504393的博客
07-29 2544
【代码】Pytorch 报错 AssertionError: If capturable=False, state_steps should not be CUDA tensors.
AssertionError: If capturable=False, state_steps should not be CUDA tensors.报错的一个解决方法。
weixin_47136182的博客
03-20 1047
我出现这个问题的原因，是使用了pytorch1.12版本（可能是版本的bug？），之后换成了1.10版本就可以了。所以如果其他的包不必须依赖1.12版本，可以尝试降低pytorch版本来解决这个问题。在跑代码的时候忽然遇到了这个问题，记录下我的解决办法。
使用MMSelfSup时，报错If capturable=False, state_steps should not be CUDA tensors.
m0_57143158的博客
09-20 195
3、我是把MMSelfSup虚拟环境删除了，重新安装了一遍。我害怕如果仅在虚拟环境中把之前的Pytorch卸载了，使用以上语句安装了新的Pytorch后，又与MMCV版本不对应，报其他错误（MMCV必须与Pytorch与cuda的版本对应。MIM安装MMCV时会根据Pytorch与CUDA的版本，自动选择合适的MMCV）。2、查了一下是Pytorch1.12.0版本的bug，无奈只能换个其他版本的Pytorch。我使用以下语句重新安装了Pytorch 1.9.0，然后就不报错了。
assert not step_t.is_cuda, “If capturable=False, state_steps should not be CUDA tensors.
最新发布
shysea2019的博客
10-21 311
将pytorch版本从1.12.0升级到1.12.1即可。
pytorch 状态字典:state_dict使用详解
09-18
今天小编就为大家分享一篇pytorch 状态字典:state_dict使用详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
FCN-pytorch-master.zip_fcn pytorch 9:9_pytorch_pytorch fcn代码_pyt
07-14
FCN的pytorch实现，实现较为简单
GAT-图注意力网络（PyTorch）:laptop:+图+:megaphone:=:red_heart:-Python开发
05-25
GAT-图注意力网络（PyTorch）:laptop:+图+:megaphone:=:red_heart:此仓库包含原始GAT论文的PyTorch实现（:link:Veličković等）。 它的目的是简化一般的GAT和GNN的玩法和学习。 目录什么是图神经网络和GAT？ 可视化...
长短期记忆网络（Long Short-Term Memory，LSTM）及其变体双向LSTM和GRU
热门推荐
陶将的博客
07-05 5万+
LSTM（Long Short-Term Memory）长短期记忆网络，是一种时间递归神经网络，适合于处理和预测时间序列中间隔和延迟相对较长的重要事件。LSTM是解决循环神经网络RNN结构中存在的“梯度消失”问题而提出的，是一种特殊的循环神经网络。
如何防止过拟合(overfitting)
陶将的博客
09-30 5万+
 为了得到一致假设而使假设变得过度复杂称为过拟合(overfitting)，过拟合表现在训练好的模型在训练集上效果很好，但是在测试集上效果差。也就是说模型的泛化能力弱。在很多问题中，我们费心费力收集到的数据集并不能穷尽所有的状态，而且一般训练得到的模型被用来预测未知数据的结果，模型虽然在训练集上效果很好，但是在实际应用中效果差，就说明这个模型训练的并不是很成功，还需要改进。就譬如下方的图像中，左边...
卷积神经网络常见架构AlexNet、ZFNet、VGGNet、GoogleNet和ResNet模型
陶将的博客
06-22 4万+
目前的常见的卷积网络结构有AlexNet、ZF Net、VGGNet、Inception、ResNet等等，接下来我们对这些架构一一详解。
TensorRT介绍
陶将的博客
03-03 4万+
TensorRT 文章目录TensorRT训练和推理的区别TensorRTTensorRT 优化和性能TensorRT 工作原理Python APIImporting TensorRT Into PythonCreating A Network Definition In PythonBuilding An Engine In PythonSerializing A Model In PythonPerforming Inference In PythonexamplesWorking With Mixed
生成对抗网络GAN---生成mnist手写数字图像示例（附代码）
陶将的博客
08-06 2万+
Ian J. Goodfellow等人于2014年在论文Generative Adversarial Nets中提出了一个通过对抗过程估计生成模型的新框架。框架中同时训练两个模型：一个生成模型（generativemodel）G，用来捕获数据分布；一个判别模型(discriminative model)D，用来估计样本来自于训练数据的概率。G的训练过程是将D错误的概率最大化。可以证明在任意函数G...
python cv2.imread 读取中文路径的图片返回为None的问题
陶将的博客
02-11 1万+
使用cv2读取图片时，输出图片形状大小时出现报错“ 'NoneType' object has no attribute shape”，后来排查发现读取图片的返回值image为None， 这就说明图片根本就没有被读取。 使用PIL读取图像，能够成功读取图片。
人脸识别之SphereFace
陶将的博客
06-12 1万+
本文主要介绍SphereFace这篇论文的基本思想
卷积神经网络：CIFAR-10训练和测试（单块GPU）
陶将的博客
08-22 1万+
 正好闲暇，将前段时间搁浅的官方cifar-10的例子运行了一遍。我只有一个GPU，所以并没有完全照搬官方的代码， 而是在此基础上，结合自己的理解，对代码进行了修改，总共有三个子文件，每个文件具体作用如下表所示。 文件 作用 cifar10_input.py 读取本地CIFAR-10的二进制文件格式，定义函数distorted_inputs获得训练数据和inputs函数获取测试数...
人脸识别之AM-Softmax
陶将的博客
06-14 1万+
论文：Additive Margin Softmax for Face Verification 在此篇论文中，作者提出了新的损失函数，AM_softmax。嗯，我们可以把AM-softmax看做是对A-softmax损失函数的改进，两者是很相似的，我刚写过A-softmax的博客，详情请点击这里。两者最大的不同之处在于：A-softmax是用margin m与相乘，而AM-softmax的ma...
pytorch cpu版AssertionError: Torch not compiled with CUDA enabled
09-27
当出现"AssertionError: Torch not compiled with CUDA enabled"错误时，说明你的PyTorch库没有启用CUDA。这个错误通常发生在你的环境没有正确安装或配置CUDA的情况下。要解决这个问题，你可以尝试以下几个步骤： 1. 确认你的计算机上已正确安装了CUDA驱动程序。你可以从NVIDIA官方网站下载和安装适用于你的显卡型号和操作系统的最新CUDA驱动程序。 2. 检查你的PyTorch版本是否与CUDA版本兼容。你可以在PyTorch官方网站上找到与你的CUDA版本兼容的PyTorch版本列表。确保你下载和安装了正确的版本。 3. 如果你使用的是GPU版本的PyTorch，确保你的显卡支持CUDA。你可以在NVIDIA官方网站上查找你的显卡型号的技术规格来确认。 4. 如果以上步骤都正确无误，但问题仍然存在，可以尝试重新安装PyTorch并启用CUDA支持。确保你按照官方文档中的指示正确配置了环境变量和编译选项。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

马鹤宁

谢谢

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值