Import intel_extension_for_pytorch as ipex

This commit is contained in:
藍+85CD 2023-04-06 12:27:22 +08:00
parent 1ced2bdd2d
commit 84b9c0ac2f

View File

@ -87,7 +87,7 @@ except:
pass
try:
import intel_extension_for_pytorch
import intel_extension_for_pytorch as ipex
if torch.xpu.is_available():
vram_state = XPU
except: