运行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'
程序运行成功
发布者:admin,转转请注明出处:http://www.yc00.com/web/1754943557a5218539.html
评论列表(0条)