Update Dockerfile and Docker Requirements to Support 50 Series Cards (Cuda 12.8+)
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.
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.
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.
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.
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).
I have a 5090 and confirmed that this also works