CUDA docker containers

Could anyone recommend a docker image to pull in order to run things with CUDA 9? I’ve got CUDA 12 installed on my Linux machine and need to run a project with PyTorch 0.4.1 version. So far I’ve found that the old CUDA containers from NVIDIA docker hub don’t seem to work (at least for me for some reason) so if anyone has a link to a place with working images with old CUDA versions you’d be my saviour.