本文档记录了在Ubuntu和Windows系统下运行超分辨算法EDSR时遇到的CUDA、Pytorch版本匹配问题。实验表明，CUDA11.0及以上版本与相应Pytorch版本才能确保程序正常运行。在Ubuntu中，即使没有安装CUDA，只要Pytorch版本正确，也能成功运行。Windows环境下则出现各种错误，可能需要进一步调试。关键在于理解硬件算力支持的CUDA版本与Pytorch版本之间的对应关系。
1.概述
最近在跑超分辨算法EDSR，中间经理了一些Bug和情况，浅浅记录一下，希望能给遇到相似问题的朋友们一些参考。在开头先说一下跑程序成功的配置：
- 系统：Ubuntu20.04
- 显卡情况：GeForce RTX 3090（驱动515.65.01），[CUDA11.0.2，cudnn8.0.5]后来发现没有这两项也可，非必要选项
- EDSR程序版本：1.2.0版本
- 虚拟环境：Pytorch1.7.0（对应CUDA11.0），cudatoolkit11.0.221 ，无cudnn
【注：按照文中的Pytorch1.2.0/1.1.0（CUDA10.0）是不可以运行的】
2.EDSR程序在不同系统下的运行
2.1.Windows系统下运行系统
由于作者官方说了在Windows下不保证可以跑通，所以在这里知识浅浅测试一下，记录如下：
2.1.1.配置pytorch1.7.1（CUDA11.0）
根据Ubuntu运行成功的经验，也安装该版本的pytorch，运行出现以下结果：
报错ImportError: cannot import name ‘_DataLoaderIter’，猜测是版本Pytorch更新之后部分函数更新导致
2.1.2.配置pytorch1.2.0（CUDA10.0）
由于pytorch1.1.0最高支持到CUDA10.0，所以只能安装该版本，运行后出现以下结果：
运行了好长时间后，若不带–save_results，可
最低0.47元/天 解锁文章
扫一扫
1992
到【灌水乐园】发言
