gpt-j-6b-gpu-docker
gpt-j-6b-gpu-docker copied to clipboard
devforth
Unavailability of "nvidia/cuda:11.1-base" Image in Dockerfile
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:
