Join GitHub today
GitHub is home to over 40 million developers working together to host and review code, manage projects, and build software together.Sign up
Use an NVIDIA base image. #3177
This yields a smaller base image with reduced complexity around NVIDIA/CUDA (we simply use the base image) but increased complexity around what version of Python we're installing (we need to manage it more manually rather than just using the base image).
This change would make it easier to run the