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

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

程序运行成功

你可能感兴趣的:(运行pytorch发生CUDA out of memory显存不足解决)