frozenleaves/accelerate
1
0
Fork 0
mirror of https://github.com/huggingface/accelerate.git synced 2025-10-20 10:03:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: torch_npu import error (#3764)

Browse Source
This commit is contained in:
Ju4tCode
2025-09-09 17:38:57 +08:00
committed by GitHub
parent a0bc36e8ed
commit 9ead94e556
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/accelerate/utils/imports.py
View File

@ -414,7 +414,12 @@ def is_npu_available(check_device=False):
if importlib.util.find_spec("torch_npu") is None:
return False
import torch_npu # noqa: F401
# NOTE: importing torch_npu may raise error in some envs
# e.g. inside cpu-only container with torch_npu installed
try:
import torch_npu # noqa: F401
except Exception:
return False
if check_device:
try: