Traceback (most recent call last): File "D:/LPRNet_Pytorch-master/LPRNet_Pytorch-master/train_LPRNet.py", line 268, in <module> train() File "D:/LPRNet_Pytorch-master/LPRNet_Pytorch-master/train_LPRNet.py", line 107, in train lprnet.load_state_dict(torch.load(args.pretrained_model)) File "D:\Anaconda\lib\site-packages\torch\serialization.py", line 608, in load return _legacy_load(opened_file, map_location, pickle_module, **pickle_load_args) File "D:\Anaconda\lib\site-packages\torch\serialization.py", line 787, in _legacy_load result = unpickler.load() File "D:\Anaconda\lib\site-packages\torch\serialization.py", line 743, in persistent_load deserialized_objects[root_key] = restore_location(obj, location) File "D:\Anaconda\lib\site-packages\torch\serialization.py", line 175, in default_restore_location result = fn(storage, location) File "D:\Anaconda\lib\site-packages\torch\serialization.py", line 151, in _cuda_deserialize device = validate_cuda_device(location) File "D:\Anaconda\lib\site-packages\torch\serialization.py", line 135, in validate_cuda_device raise RuntimeError('Attempting to deserialize object on a CUDA ' RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
时间: 2023-08-27 10:05:30 浏览: 59
这个错误是因为你在尝试从一个CUDA设备上加载模型,但是当前的机器上没有GPU。你需要使用`torch.load`函数时传入`map_location=torch.device('cpu')`的参数来将模型加载到CPU上。 例如:
```python
lprnet.load_state_dict(torch.load(args.pretrained_model, map_location=torch.device('cpu')))
```
相关问题
Traceback (most recent call last): File "D:/LPRNet_Pytorch-master/LPRNet_Pytorch-master/train_LPRNet.py", line 268, in <module> train() File "D:/LPRNet_Pytorch-master/LPRNet_Pytorch-master/train_LPRNet.py", line 102, in train lprnet.to(device) File "D:\Anaconda\lib\site-packages\torch\nn\modules\module.py", line 899, in to return self._apply(convert) File "D:\Anaconda\lib\site-packages\torch\nn\modules\module.py", line 570, in _apply module._apply(fn) File "D:\Anaconda\lib\site-packages\torch\nn\modules\module.py", line 570, in _apply module._apply(fn) File "D:\Anaconda\lib\site-packages\torch\nn\modules\module.py", line 593, in _apply param_applied = fn(param) File "D:\Anaconda\lib\site-packages\torch\nn\modules\module.py", line 897, in convert return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking) File "D:\Anaconda\lib\site-packages\torch\cuda\__init__.py", line 208, in _lazy_init raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not compiled with CUDA enabled
这个错误是因为你的torch没有启用CUDA。在训练LPRNet时需要使用GPU加速,因此需要启用CUDA。你可以检查一下你的torch是否安装了CUDA版本,并且在代码中使用了正确的版本。如果你的torch没有安装CUDA版本,你需要重新安装一个CUDA版本的torch。如果你已经安装了CUDA版本的torch,你可以尝试在代码中添加以下代码来启用CUDA:
```
import torch
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
```
然后在将模型转移到设备时使用:
```
lprnet.to(device)
```
这应该会解决你的问题。
Traceback (most recent call last): File "C:\Users\31225\Desktop\MVSNet_pytorch-master\train.py", line 16, in <module> from utils import * File "C:\Users\31225\Desktop\MVSNet_pytorch-master\utils.py", line 2, in <module> import torchvision.util
这个错误是因为在你的代码中,`utils.py`文件中尝试导入了`torchvision.util`,但是该模块在`torchvision`库中不存在。你需要检查你的代码,并确认你要导入的模块的名称是否正确。如果你想使用`torchvision`库,请确保已经正确安装该库。你可以尝试通过在终端中运行以下命令来安装它:
```
pip install torchvision
```
如果你已经安装了该库,请检查你的安装是否正确,并且版本是否与你的代码兼容。