onnx_runtime_cpp
onnx_runtime_cpp copied to clipboard
Updated ubuntu2004_gpu Dockerfile specs
Purpose of Pull Request :bookmark:
This is to update the base image used as well as add COPY instructions to the file ubuntu2004_gpu.dockerfile under directory dockerfiles, in order to avoid compilation errors such as missing bash scripts as well as docker base image.
As of 21st September 2024, realized that Nvidia has removed the now deprecated nvidia/cuda:11.4.0-cudnn8-devel-ubuntu20.04 from their DockerHub.
Summary :books:
- Updated base image to
nvidia/cuda:11.4.3-cudnn8-devel-ubuntu20.04instead which is still available. - Added COPY instructions to ensure the custom bash scripts for installing dependencies are present during docker image build process.