Install and run CUDA 11.3 in docker container

Hi everyone,

I am trying to run my code for local testing but I didn't manage to install CUDA 11.3 in my docker container. I followed this tutorial , it installs CUDA 10.2 succesfully, however I have an RTX 3090 which requires cuda version 11.3. Anyone knows how to fix this?

Kind regards,

Dear aminey,

we had the same problem and decided to try and create our own solution. This works with building locally but we haven't tested it for the official submission yet. Hope it helps anyway. Feel free to ask any questions that arise.

https://gist.github.com/Moxnie/c7a2388a57cf854f0a17469049ebc4b3 Regards, TS

Thank you TS_UKE for sharing your dockerfile :) I will try it out and share my updates.

Regards,

UPDATE: Your script worked perfectly thanks :). I also tried to change from the pytorch image to python 3.9 image and then install pytorch/torchvision/torchaudio with cuda 11.3 separatelty.

FROM python:3.9 RUN python -m pip install torch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 --extra-index-url https://download.pytorch.org/whl/cu113