时间戳

首页 > 解决方案 > 无法从 Pytorch 转换 MobiletNet_v1

python - 无法从 Pytorch 转换 MobiletNet_v1

问题描述

当我尝试为变式自动编码器运行我的代码时，我收到此错误：

test_and_train.py:None (test_and_train.py)
test_and_train.py:102: in <module>
    model_net = VoxceptionNet(n_classes=40).to(device)
..\..\..\Anaconda3\envs\lib\site-packages\torch\nn\modules\module.py:381: in to
    return self._apply(convert)
..\..\..\Anaconda3\envs\lib\site-packages\torch\nn\modules\module.py:187: in _apply
    module._apply(fn)
..\..\..\Anaconda3\envs\lib\site-packages\torch\nn\modules\module.py:193: in _apply
    param.data = fn(param.data)
..\..\..\Anaconda3\envs\lib\site-packages\torch\nn\modules\module.py:379: in convert
    return t.to(device, dtype if t.is_floating_point() else None, non_blocking)
E   RuntimeError: CUDA error: invalid device ordinal

标签： python

解决方案

如果在不同数量的 Cuda 设备上训练预训练模型，您可能会遇到该错误。

你可以参考github上的这个问题，它可能会对你有所帮助。

推荐阅读