{"library":"torch-npu","type":"library","category":null,"description":"torch-npu is a PyTorch extension that serves as an NPU bridge, adapting the Ascend Neural Network Processing Unit (NPU) to the PyTorch framework. It enables developers to leverage the powerful computational capabilities of Huawei Ascend AI Processors for deep learning training and inference within the PyTorch ecosystem. The current version is 2.9.0, with regular updates aligning with PyTorch releases and Ascend software stacks.","language":"python","status":"active","version":"2.5.1","tags":["pytorch","npu","ascend","hardware-acceleration","deep-learning","huawei"],"last_verified":"Sat May 23","install":[{"cmd":"pip install pyyaml setuptools\npip install torch==2.9.0\npip install torch-npu==2.9.0","imports":["import torch_npu","import torch","torch.npu.is_available()","torch.npu.set_device('npu:0')"]},{"cmd":"pip install pyyaml setuptools\npip install torch==2.9.0+cpu --index-url https://download.pytorch.org/whl/cpu\npip install torch-npu==2.9.0","imports":[]},{"cmd":"pip install pyyaml setuptools\npip install torch==2.9.0\npip install torch-npu==2.9.0","imports":[]}],"homepage":"https://gitcode.com/ascend/pytorch","github":null,"docs":null,"changelog":null,"pypi":"https://pypi.org/project/torch-npu/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":10,"avg_install_s":26.8,"avg_import_s":null,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/torch-npu/compatibility"}}