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