- 博客(2)
- 收藏
- 关注
原创 pytorch出现RuntimeError: CUDA out of memory.--------MDSR
在运行MDSR程序时,分别两次出现CUDA out of memory.第一次是在数据训练的过程中,由于Batch_size设置过大,导致内存不足,将其设置较小即可解决。第二次是在测试过程中,由于是直接导入训练好的模型,不需要再次进行反向传播。因此在进行前向传播之前加入with torch.no_grad():即可解决。即with torch.no_grad(): ...
2019-08-08 08:52:26
3383
原创 有关python中出现ImportError: DLL load failed: 找不到指定的程序
百度了一天,试过了若干方法,导入的时候老是出现ImportError: DLL load failed: 找不到指定的程序。最后的解决方法是:安装的Numpy应该是 numpy+mkl,我之前一直安装的是numpy-1.16.4+vanilla-cp36-cp36m-win_amd64,被自己蠢哭,感谢一位大神的强调。...
2019-06-27 16:44:49
928
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人