Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

Provide pre-built wheels with CUDA support. #1143

Merged
merged 6 commits into from
Jul 17, 2024

Conversation

csukuangfj
Copy link
Collaborator

You can find them at
https://k2-fsa.github.io/sherpa/onnx/cuda.html

Screenshot 2024-07-17 at 22 57 13

Note that only linux x64 and windows x64 are supported.


Usage

pip install sherpa-onnx==1.10.16+cuda -f https://k2-fsa.github.io/sherpa/onnx/cuda.html

The log is

Looking in links: https://k2-fsa.github.io/sherpa/onnx/cuda.html
Collecting sherpa-onnx==1.10.16+cuda
  Downloading https://huggingface.co/csukuangfj/sherpa-onnx-wheels/resolve/main/cuda/1.10.16/sherpa_onnx-1.10.16%2Bcuda-cp310-cp310-linux_x86_64.whl (183.3 MB)
     ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ 183.3/183.3 MB 3.8 MB/s eta 0:00:00
Installing collected packages: sherpa-onnx
Successfully installed sherpa-onnx-1.10.16+cuda
import sherpa_onnx
print(sherpa_onnx.__version__)

should print

1.10.16+cuda

@csukuangfj csukuangfj merged commit 4198d9a into k2-fsa:master Jul 17, 2024
17 checks passed
@csukuangfj csukuangfj deleted the wheel-cuda branch July 17, 2024 14:59
XiaYucca pushed a commit to XiaYucca/sherpa-onnx that referenced this pull request Jan 9, 2025
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant