admin 管理员组

文章数量: 1184232

运行pytorch发生显存不足解决

版本：
python：3.7
pytorch：1.2
cuda：10.1

做pytorch迁移学习时发生显存不足事件

也就是CUDA out of memory

使用nvidia-smi查看gpu信息（需要先把C:\Program Files\NVIDIA Corporation\NVSMI添加到Path）

发现事实上我的显存是很足的

据说可能是tensorflow与pytorch冲突

尝试卸载tensorflow-gpu

无效

在程序中输入一下代码

import os
os.environ['CUDA_VISIBLE_DEVICES]='2，3'

程序运行成功

本文标签： 显存 发生 pytorch memory CUDA

版权声明：本文标题：运行pytorch发生CUDA out of memory显存不足解决 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://roclinux.cn/b/1754943561a3052921.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。