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

运行pytorch发生显存不足解决

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

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

也就是

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

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

据说可能是tensorflow与pytorch冲突

尝试卸载tensorflow-gpu

无效

在程序中输入一下代码

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

程序运行成功

发布了6 篇原创文章 · 获赞 0 · 访问量 319

猜你喜欢

转载自blog.csdn.net/qq_43452745/article/details/102593648