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