为了复现一篇用神经网络处理图像的论文,第一次接触到Pytorch,于是就随手百度了一篇安装教程https://blog.youkuaiyun.com/weixin_43012220/article/details/83786637,万万没想到,这就是噩梦的开始……
运行代码的时候发现了非常严重的两个问题:1. 运行时间长 2.输出图像效果极差
根据以往使用waifu2x的经验,我立刻明白了这是大概是因为代码处理图像时没有用GPU而只用了CPU。重新运行代码,同时打开任务管理器的“性能”来查看设备使用情况,果不其然CPU几乎满载,GPU根本没有用上。
https://blog.youkuaiyun.com/nima1994/article/details/83001910,通过这个篇博客所讲内容,在代码中添加相关代码,所得结果全部都为无法使用GPU的表现。比如,将torch.cuda.is_available() 的返回值print出来,一直是False。
在网上查了很久很久,什么要额外再安装nividia cudnn之类的,试过了基本都没用,整个人完全陷入自闭状态……
后来又翻到了一两篇博客
https://blog.youkuaiyun.com/qq_38410428/article/details/82965499