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

Support Python 3.12, Torch 2.4, CUDA 12.4 #39

Open
C0rn3j opened this issue Aug 12, 2024 · 3 comments
Open

Support Python 3.12, Torch 2.4, CUDA 12.4 #39

C0rn3j opened this issue Aug 12, 2024 · 3 comments
Labels
enhancement New feature or request

Comments

@C0rn3j
Copy link

C0rn3j commented Aug 12, 2024

Torch 2.3.x only supports CUDA 12.1
Torch 2.4, which came out 3 weeks ago (current Docker builds are older), supports CUDA 12.4 and also Python 3.12
Upcoming Torch 2.4.1 seems to support CUDA 12.5.

I see the dependencies simply specify latest torch, so maybe just rebuilding the current image and pushing an update would be enough?

EDIT: Apparently I am ignorant for thinking you need to match host CUDA version to the torch version, newer CUDA on host seems to work fine with older CUDA on apps and was running into a Docker documentation issue with not being able to get it to run, but updating the image would still be nice.

@C0rn3j C0rn3j changed the title Support Torch 2.4 Support Torch 2.4 for CUDA 12.4 Aug 12, 2024
@matatonic
Copy link
Owner

Is there any issue with older torch or CUDA here or just a request for updated versions? I update as far as I can normally, but some things have tough dependencies.

@C0rn3j
Copy link
Author

C0rn3j commented Aug 15, 2024

Not really,
I realized you can have a newer CUDA on host and just run with older CUDA on the actual apps too late, otherwise I would have not made this issue.

2.4.0 does add Python 3.12 compatiblity though, which some people would probably welcome as modern distros now ship with that.

@matatonic matatonic changed the title Support Torch 2.4 for CUDA 12.4 Support Python 3.12, Torch 2.4, CUDA 12.4 Aug 15, 2024
@matatonic
Copy link
Owner

I'll leave this open because it should be done eventually, but for now it's not going to happen, I've already tried upgrading to torch 2.3 and had a host of problems.

@matatonic matatonic added the enhancement New feature or request label Aug 27, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

No branches or pull requests

2 participants