Skip to content

Update Dockerfile and Docker Requirements to Support 50 Series Cards (Cuda 12.8+) - #4102

Open
jason-e-gross wants to merge 4 commits into
lllyasviel:mainfrom
jason-e-gross:main
Open

Update Dockerfile and Docker Requirements to Support 50 Series Cards (Cuda 12.8+)#4102
jason-e-gross wants to merge 4 commits into
lllyasviel:mainfrom
jason-e-gross:main

Conversation

@jason-e-gross

@jason-e-gross jason-e-gross commented Sep 20, 2025

Copy link
Copy Markdown

I made these two changes to the Dockerfile and the docker_requrirements.txt, and was able to run the docker build and successfully make a docker image that works on 50 series cards.

Tested it on my Nvidia 5060 TI 16GB and it works.

@jason-e-gross jason-e-gross changed the title Update for 12.8/13 Update Dockerfile and Docker Requirements to Support 50 Series Cards (Cuda 12.8+) Sep 20, 2025
@jason-e-gross

Copy link
Copy Markdown
Author

As an FYI - I was using it on a Linux Mint OS install, running the latest available drivers (that had CUDA 13 shown in the nvidia-smi command). The machine has both a 4070 and a 5060 in it - and I targeted the 5060 specifically in my docker compose using this image. All worked great.

@jason-e-gross

jason-e-gross commented Sep 20, 2025

Copy link
Copy Markdown
Author

20250920_175634
image

Sorry about the crappy screenshots, but the machine it runs on is on a secured VLAN so no easy way in. These are screenshots from the machine running the docker image.

@mashb1t

mashb1t commented Sep 25, 2025

Copy link
Copy Markdown
Collaborator

Hey @jason-e-gross, thanks for the update, much appreciated!

I sadly can't test the change as i do not have a 5000 series card.

Can anybody else please confirm if these changes are working without issues? Will merge afterwards.

@jason-e-gross

jason-e-gross commented Oct 1, 2025

Copy link
Copy Markdown
Author

Hey @jason-e-gross, thanks for the update, much appreciated!

I sadly can't test the change as i do not have a 5000 series card.

Can anybody else please confirm if these changes are working without issues? Will merge afterwards.

Just an FYI - I've also tested it by targeting the 4070 and these changes work with them still, also - so seems solid for both old and new. Just thought I'd throw that in there, this isn't a breaking change update.

I built the docker image, I suppose I could probably push it somewhere folks could try the image directly. I'll look into it this week, if I can (otherwise, building the docker image is pretty straightforward and painless).

@kobaltz

kobaltz commented Oct 9, 2025

Copy link
Copy Markdown

I have a 5090 and confirmed that this also works

Refactor Dockerfile to improve readability and maintainability. Added specific package versions for compatibility and organized RUN commands.
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.

4 participants