Unavailability of "nvidia/cuda:11.1-base" Image in Dockerfile

[GoutamVerma]

Issue Description:

The current Dockerfile is encountering an error during the build process because it cannot find the "nvidia/cuda:11.1-base" image at the specified path. As a resolution, it is recommended to use the "nvidia/cuda:11.0.3-base-${DIST}" images instead, where "DIST" can be one of the following: ubuntu20.04, ubuntu18.04, centos7, among others. The list of available tags for the "nvidia/cuda" image can be found at this link.

Suggested Solution:

To resolve this issue, the Dockerfile should be updated to use the appropriate "nvidia/cuda:11.0.3-base-${DIST}" image instead of "nvidia/cuda:11.1-base". The correct image can be chosen based on the target distribution ("DIST") as required. Follow the steps below to modify the Dockerfile:

Identify the desired "DIST" value (e.g., ubuntu20.04, ubuntu18.04, centos7).
Update the "FROM" instruction in the Dockerfile to use the recommended image format: