常规设置

1.手动清空内存

查找占用显存的进程,然后kill 进程即可

sudo fuser -v /dev/nvidia* #查找占用GPU资源的PID
2. 设置显卡
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
posted @ 2019-04-03 14:31  烨然2333  阅读(180)  评论(0编辑  收藏  举报