Update config.py
This commit is contained in:
parent
deb756bf1b
commit
b83939ba04
23
config.py
23
config.py
@ -148,39 +148,28 @@ class Config:
|
|||||||
x_max = 32
|
x_max = 32
|
||||||
if self.dml:
|
if self.dml:
|
||||||
print("use DirectML instead")
|
print("use DirectML instead")
|
||||||
|
if(os.path.exists("runtime\Lib\site-packages\onnxruntime\capi\DirectML.dll")==False):
|
||||||
try:
|
try:
|
||||||
os.rename(
|
os.rename("runtime\Lib\site-packages\onnxruntime", "runtime\Lib\site-packages\onnxruntime-cuda")
|
||||||
"runtime\Lib\site-packages\onnxruntime",
|
|
||||||
"runtime\Lib\site-packages\onnxruntime-cuda",
|
|
||||||
)
|
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
try:
|
try:
|
||||||
os.rename(
|
os.rename("runtime\Lib\site-packages\onnxruntime-dml", "runtime\Lib\site-packages\onnxruntime")
|
||||||
"runtime\Lib\site-packages\onnxruntime-dml",
|
|
||||||
"runtime\Lib\site-packages\onnxruntime",
|
|
||||||
)
|
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
import torch_directml
|
import torch_directml
|
||||||
|
|
||||||
self.device = torch_directml.device(torch_directml.default_device())
|
self.device = torch_directml.device(torch_directml.default_device())
|
||||||
self.is_half = False
|
self.is_half = False
|
||||||
else:
|
else:
|
||||||
if self.instead:
|
if self.instead:
|
||||||
print(f"use {self.instead} instead")
|
print(f"use {self.instead} instead")
|
||||||
|
if(os.path.exists("runtime\Lib\site-packages\onnxruntime\capi\onnxruntime_providers_cuda.dll")==False):
|
||||||
try:
|
try:
|
||||||
os.rename(
|
os.rename("runtime\Lib\site-packages\onnxruntime", "runtime\Lib\site-packages\onnxruntime-dml")
|
||||||
"runtime\Lib\site-packages\onnxruntime",
|
|
||||||
"runtime\Lib\site-packages\onnxruntime-dml",
|
|
||||||
)
|
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
try:
|
try:
|
||||||
os.rename(
|
os.rename("runtime\Lib\site-packages\onnxruntime-cuda", "runtime\Lib\site-packages\onnxruntime")
|
||||||
"runtime\Lib\site-packages\onnxruntime-cuda",
|
|
||||||
"runtime\Lib\site-packages\onnxruntime",
|
|
||||||
)
|
|
||||||
except:
|
except:
|
||||||
pass
|
pass
|
||||||
return x_pad, x_query, x_center, x_max
|
return x_pad, x_query, x_center, x_max
|
||||||
|
Loading…
x
Reference in New Issue
Block a user