Merge pull request #155 from ShivamKumar2002/patch-1
Fix GPU not being used even when it available
This commit is contained in:
commit
6416d4ebe8
@ -1,3 +1,4 @@
|
||||
import torch
|
||||
import onnxruntime
|
||||
|
||||
use_gpu = False
|
||||
|
Loading…
Reference in New Issue
Block a user