Mang Sun
Mang Sun
torch1.12.1
你可以参考这篇文章https://blog.csdn.net/literacy_wang/article/details/116006259?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522170515576516800211568640%2522%252C%2522scm%2522%253A%252220140713.130102334..%2522%257D&request_id=170515576516800211568640&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~all~sobaiduend~default-2-116006259-null-null.142^v99^pc_search_result_base7&utm_term=EDSR%E4%BB%A3%E7%A0%81%E5%A4%8D%E7%8E%B0&spm=1018.2226.3001.4187
看一下你是否是下载的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里面直接使用绝对路径,将你的测试集放进去