运行pytorch发生CUDA out of memory显存不足解决

运行pytorch发生显存不足解决版本:python:3.7pytorch:1.2cuda:10.1做pytorch迁移学习时发生显存不足事件

运行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)

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

据说可能是tensorflowpytorch冲突

尝试卸载tensorflow-gpu

无效

在程序中输入一下代码

import os
os.environ['CUDA_VISIBLE_DEVICES]='23'

程序运行成功

发布者:admin,转转请注明出处:http://www.yc00.com/web/1754943557a5218539.html

相关推荐

发表回复

评论列表(0条)

  • 暂无评论

联系我们

400-800-8888

在线咨询: QQ交谈

邮件:admin@example.com

工作时间:周一至周五,9:30-18:30,节假日休息

关注微信