robinwitch commited on
Commit
e410fbf
·
1 Parent(s): 72a482d
Files changed (1) hide show
  1. system_utils.py +1 -1
system_utils.py CHANGED
@@ -1,7 +1,7 @@
1
  import pynvml
2
 
3
  def get_gpt_id():
4
- # return "1"
5
  pynvml.nvmlInit()
6
  gpu_indices = []
7
  device_count = pynvml.nvmlDeviceGetCount()
 
1
  import pynvml
2
 
3
  def get_gpt_id():
4
+ return "0"
5
  pynvml.nvmlInit()
6
  gpu_indices = []
7
  device_count = pynvml.nvmlDeviceGetCount()