fix gpu issues for some envs
This commit is contained in:
@@ -1,3 +1,4 @@
|
|||||||
|
import torch
|
||||||
import onnxruntime
|
import onnxruntime
|
||||||
|
|
||||||
use_gpu = False
|
use_gpu = False
|
||||||
|
|||||||
Reference in New Issue
Block a user