FlagEmbedding调用显卡问题
#4
by labniz - opened
os.environ["CUDA_VISIBLE_DEVICES"] = "[5,6]"？ 还是 os.environ["CUDA_VISIBLE_DEVICES"] = "5,6"？呀
作者您好，不太清楚通过FlagEmbedding使用gpu的方式，输出会显示说
----------using 8*GPUs----------
但看显卡情况是没有用到
labniz changed discussion status to closed
labniz changed discussion status to open
labniz changed discussion title from os.environ["CUDA_VISIBLE_DEVICES"] to os.environ["CUDA_VISIBLE_DEVICES"]调用显卡问题
labniz changed discussion title from os.environ["CUDA_VISIBLE_DEVICES"]调用显卡问题 to FlagEmbedding调用显卡问题
是os.environ["CUDA_VISIBLE_DEVICES"] = "5,6"