search

MaoXinn / SEU

30 stars 4 forks source link

请问使用sinkgorn的时候怎么使用gpu呢 #8

Closed zulihit closed 1 year ago

zulihit commented 1 year ago

我尝试了设置os.environ["CUDA_VISIBLE_DEVICES"] = "0"，但是运算时间好像并没有什么变化 貌似Sinkhorn的计算时间完全与迭代次数相关

MaoXinn commented 1 year ago

额，我这边没有这个问题。是不是TensorFlow或者cuda版本的问题？