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"