-
Notifications
You must be signed in to change notification settings - Fork 33
Add CUDA build + packaging workflow #672
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
Add CUDA build + packaging workflow #672
Conversation
- name: Set target dockerfile | ||
run: | | ||
DOCKERFILE="Dockerfile.release" | ||
if [[ "${{ inputs.cuda }}" == true ]]; then |
Check failure
Code scanning / SonarCloud
GitHub Actions should not be vulnerable to script injections High
|
This is working (I've built the image and run it on GPUs). Merging into my branch, I think we should polish the API a bit. |
Description
Added
Dockerfile.cuda
specifically for CUDA build and updated packaging workflows to use it.A manual build seems to show CUDA as available, e.g.:
12.6
2.6.0
TODO:
torch-*
install to use prebuilt packages for whichever version oftorch
we end up building with (otherwise building takes forever, as these packages seem to require building from source).torch
version - current version used in MAGE is2.6.0
; current stable version is2.8.0
- recommendation: use2.6.0
untildgl
is removed.13.0
, but the latest supported bytorch==2.8.0
is12.9
; recommendation: use version12.6
as it is supported by all versions oftorch
from2.6.0
to2.8.0
, and it still has support for older drivers (>=525
to<580
) and GPUs (Pascal, Maxwell).Dockerfile.release
Pull request type
Related issues
Delete if this PR doesn't resolve any issues. Link the issue if it does.
######################################
Reviewer checklist (the reviewer checks this part)
Module/Algorithm
Documentation checklist