Pytorch GPU显存充足却显示out of memory的解决方式
今天在测试一个pytorch代码的时候显示显存不足，但是这个网络框架明明很简单，用CPU跑起来都没有问题，GPU却一直提示out of memory.
在网上找了很多方法都行不通，最后我想也许是pytorch版本的问题，原来我的pytorch版本是0.4.1，于是我就把这个版本卸载，然后安装了pytorch1.1.0，程序就可以神奇的运行了，不会再有OOM的提示了。虽然具体原因还不知道为何，这里还是先mark一下，具体过程如下：
卸载旧版本pytorch：
conda uninstall pytorch
安装pytorch1.1.0，按照官网上的办法，我的CUDA版本是9.0：
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
大功告成!
以上这篇Pytorch GPU显存充足却显示out of memory的解决方式就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持小牛知识库。
-
本文向大家介绍tensorboard 可以显示graph,却不能显示scalar的解决方式，包括了tensorboard 可以显示graph,却不能显示scalar的解决方式的使用技巧和注意事项，需要的朋友参考一下 今天照着样例搞了下tensorboard，发现自己无法显示scalar，而graph却可以正常显示。 出现这种情况就说明，tensorfboard已经正确读取了指定目录下的数据，只是数
-
本文向大家介绍Apache无法自动跳转却显示目录的解决方法，包括了Apache无法自动跳转却显示目录的解决方法的使用技巧和注意事项，需要的朋友参考一下 Apache无法自动跳转却显示目录这是因为Apache服务器下conf/httpd.conf没有配置好，其默认是不跳转，直接显示目录 首先，我们需要禁止Apache直接显示目录。 方法在搜索Options Indexes FollowSymLink
-
本文向大家介绍tensorboard显示空白的解决，包括了tensorboard显示空白的解决的使用技巧和注意事项，需要的朋友参考一下 ubuntu 14.04 + python3.4 + chrome， 在浏览器中查看tensorboard， 发现出了graph，其他的数据都是空白。 通过分析，发现js中如下一些错误 Uncaught SyntaxError: Block-scoped decl
-
本文向大家介绍python中ImageTk.PhotoImage()不显示图片却不报错问题解决，包括了python中ImageTk.PhotoImage()不显示图片却不报错问题解决的使用技巧和注意事项，需要的朋友参考一下 发现问题 今天在使用ImageTk.photoImage()显示图片时，当把包含该函数放在自定义函数里时，不能正常显示，移到函数为又可正常显示，所以想到可能是变量不是全局性的缘
-
本文向大家介绍基于pycharm导入模块显示不存在的解决方法，包括了基于pycharm导入模块显示不存在的解决方法的使用技巧和注意事项，需要的朋友参考一下 最近，同级或者不同级目录下，导入某个模块，显示不存在，可明明存在，百度找了好多没找到，试了 这方法还不行 解决办法：pycharm不会将当前文件目录自动加入自己的sourse_path。右键make_directory as-->sources
-
本文向大家介绍vue2.0 路由不显示router-view的解决方法，包括了vue2.0 路由不显示router-view的解决方法的使用技巧和注意事项，需要的朋友参考一下 今天学习vue2.0 的 router-view , 爆出的错误不计其数，不知道哪位大神写的router-view,配置的参数竟然不是router，而是routes =_=|| 这是花了一上午整出来的main.js，还有网上