updated skeleton with true data
This commit is contained in:
4
worker/core/gpu_detect.py
Normal file
4
worker/core/gpu_detect.py
Normal file
@ -0,0 +1,4 @@
|
||||
import torch
|
||||
|
||||
def device():
|
||||
return 'cuda' if torch.cuda.is_available() else 'cpu'
|
||||
Reference in New Issue
Block a user