Coverage for gamdpy/misc/select_gpu.py: 75%

1from numba import cuda 

2 

3def select_gpu(): 

4 """ Select the first available GPU (necessary on some multi-gpu systems). """ 

5 for i in range(len(cuda.gpus)): 

6 try: 

7 cuda.select_device(i) 

8 print(f"Using device {i}") 

9 break 

10 except Exception: 

11 pass 

12 return 

13 

14if __name__ == "__main__": 

15 select_gpu()