大家好,又见面了,我是你们的朋友全栈君。
代码中添加:
import os
os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID”
os.environ[“CUDA_VISIBLE_DEVICES”] = “0” //这里是自己的GPU编号,我是”0″,原文是”8″.
发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/148396.html原文链接:https://javaforall.cn
本文分享自 作者个人站点/博客 前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!