Running on cuda 121 #352
rzezagar
started this conversation in
Show and tell
Running on cuda 121
#352
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
ONNX can be installed for higher cuda versions as well, instead of cu118 recommended here. Follow this guide:
https://onnxruntime.ai/docs/get-started/with-python
pip install onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/Beta Was this translation helpful? Give feedback.
All reactions