The current Dockerfile doesn't support GPUs since it doesn't install the necessary NVIDIA libraries. Adding a Dockerfile-gpu to the repository will make it much easier for users with GPU support to use NDK.
Describe your proposed implementation
Replace the final python image layer with the provided NVIDIA hpc layer.
Describe possible alternatives
Install the required NVIDIA libraries directly into the python layer.
Describe the new feature or enhancement
The current Dockerfile doesn't support GPUs since it doesn't install the necessary NVIDIA libraries. Adding a
Dockerfile-gpu
to the repository will make it much easier for users with GPU support to use NDK.Describe your proposed implementation
Replace the final python image layer with the provided NVIDIA hpc layer.
Describe possible alternatives
Install the required NVIDIA libraries directly into the python layer.