大家好,又见面了,我是你们的朋友全栈君。
代码中添加:
import os
os.environ[“CUDA_DEVICE_ORDER”] = “PCI_BUS_ID”
os.environ[“CUDA_VISIBLE_DEVICES”] = “0” //这里是自己的GPU编号,我是”0″,原文是”8″.
发布者:全栈程序员-用户IM,转载请注明出处:https://javaforall.cn/148396.html原文链接:https://javaforall.cn
【正版授权,激活自己账号】: Jetbrains全家桶Ide使用,1年售后保障,每天仅需1毛
【官方授权 正版激活】: 官方授权 正版激活 支持Jetbrains家族下所有IDE 使用个人JB账号...