AttributeError: module 'torch.backends' has no attribute 'mps'

[Liiiiaictx]

我用作者预训练的模型去测试，出现 (base) root@autodl-container-815011ba52-c31a7ebc:~/autodl-tmp/EDSR-PyTorch/src# python main.py --data_test Demo --scale 2 --n_resblocks 32 --n_feats 256 --res_scale 0.1 --pre_train ../experiment/model/EDSR_x2.pt --test_only --save_results Making model... Traceback (most recent call last): File "main.py", line 33, in main() File "main.py", line 23, in main _model = model.Model(args, checkpoint) File "/root/autodl-tmp/EDSR-PyTorch/src/model/init.py", line 24, in init if torch.backends.mps.is_available(): AttributeError: module 'torch.backends' has no attribute 'mps' 请问这个问题该怎么解决？

[sunmang]

看一下你是否是下载的gpu版本的torch，查看一些torch的版本和cuda的版本是否匹配.如果cuda版本过低，也可以尝试升级一些cuda版本，比如升级到11.6。如果你想测试基本集Set5，可以使用这个命令python main.py --data_test Set5 --data_range 801-900 --scale 4 --n_resblocks 32 --n_feats 256 --res_scale 0.1 --pre_train download --test_only --self_ensemble。如果你使用上述命令报错的话，基本是你测试集的位置没放对，可以在option.py里面直接使用绝对路径，将你的测试集放进去