RCAN模型源码调试总结,问题已解决

环境要求,几乎是所有问题的根源,torch 0.4.1,torchvision 0.2.1,python3.6,如果是租服务器的,显卡只能选一种Tesla V100,切记!

笔者尝试过其他显卡,在环境配置和作者完全一样,包括cuda都改成了8.0,但是一直不停报错,一个error解决掉还有新的error,无穷尽也,尤其是这个错误

RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED

 查阅了很多答案,大部分都说是因为cuda,cudnn,pytorch,python,torchvision的版本问题,结果我改的和作者一模一样了,也还是报相同的错误,直到!我租服务器的时候选了显卡v100!,package的版本就按作者给的requiremens.txt里面写的,cuda用的默认的11.2,没有和torch匹配,但是并不影响!依然运行起来了!而且这中间除了按我的需要修改了option文件里个别参数以外,没有修改任何地方的代码,都是按作者原本的来的,依然没有任何报错!所以这个代码能够运行的关键就是显卡的型号,其他都无关紧要,就算改了当下也会有新的错误。

如果你在运行中间遇到什么问题,可以留言,大家一起讨论。

在云服务器下下载低版本的torch方法如下

先新建一个环境,并激活,然后下载包

conda create -n RCAN python=3.6
conda activate RCAN
conda install pytorch==0.4.1 torchvision==0.2.1 cudatoolkit=9.2 -c pytorch 
conda install matplotlib
conda install scikit-image==0.15.0 scipy==1.4.1 tqdm==4.19.9 

后续,报错:AttributeError: module 'scipy.misc' has no attribute 'imsave'

misc.imsave('{}{}.png'.format(filename, p), ndarr) #把这句改为
imageio.imwrite('{}{}.png'.format(filename, p), ndarr)
#因为版本问题,第一句scipy.misc不支持函数imsave了,导入新的包imageio代替

你可能感兴趣的:(python,深度学习,开发语言)