- 博客(2)
- 收藏
- 关注
RSS订阅原创 ValueError:only one element tensors can be converted to Python scalars 解决方法
disp_image部分是一个维度为[1,4]的tensors。因此不可将元素逐个读取出来。，具体来说为tensor有些维度为【1，4】，有些为【4，】也就是总维度为2/1维。此时，已经把数据处理成可用状态，但是数据集中的tensor维度不一致，将tensor转换成numpy, 再将numpy转换为list。经过此步骤可以使得tensor都成为【4，】如果第一维为1，则进行降维操作，否则。将2维的部分降为1维。
2022-11-01 17:12:20
1608
1
原创 THCudaCheck FAIL file=torch/csrc/cuda/Module.cpp line=xx error=xx : unknown error BUG解决方案
出现此BUG后，我先是观察到torch.cuda.is_available()返回false，于是考虑将torch/torchvision/cudatoolkit等一系列工具包卸载重新安装。如果你能够解决可以参考以下，使用conda/pip uninstall pytorch卸载。这是我在测试一个模型时出现的错误。简而言之，是在运行途中我将电脑待机，等电脑重新打开后发现代码结束运行，并且再次运行出现此错误。这个错误应该是TORCH本身的一种BUG，即待机就会出现。简单粗暴，但是这个BUG被解决。
2022-11-01 11:14:31
308
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人