Skip to content
New issue

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

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

Already on GitHub? Sign in 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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant