diff --git a/deployments/container/Dockerfile.ubi9 b/deployments/container/Dockerfile.ubi9 index 0803436..04b2f02 100644 --- a/deployments/container/Dockerfile.ubi9 +++ b/deployments/container/Dockerfile.ubi9 @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -FROM nvcr.io/nvidia/cuda:12.6.1-base-ubi9 as build +FROM nvcr.io/nvidia/cuda:12.6.2-base-ubi9 as build SHELL ["/bin/bash", "-c"] @@ -58,7 +58,7 @@ RUN go install -ldflags "-extldflags=-Wl,-z,lazy -s -w" \ github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk@${NVIDIA_CTK_VERSION} \ && cp ${GOPATH}/bin/nvidia-ctk /artifacts/nvidia-ctk -FROM nvcr.io/nvidia/cuda:12.6.1-base-ubi9 +FROM nvcr.io/nvidia/cuda:12.6.2-base-ubi9 # Remove CUDA libs(compat etc) in favor of libs installed by the NVIDIA driver RUN dnf remove -y cuda-* diff --git a/deployments/container/Dockerfile.ubuntu b/deployments/container/Dockerfile.ubuntu index c8e5105..b6be81d 100644 --- a/deployments/container/Dockerfile.ubuntu +++ b/deployments/container/Dockerfile.ubuntu @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -FROM nvcr.io/nvidia/cuda:12.6.1-base-ubuntu20.04 as build +FROM nvcr.io/nvidia/cuda:12.6.2-base-ubuntu20.04 as build SHELL ["/bin/bash", "-c"] @@ -58,7 +58,7 @@ RUN go install -ldflags "-extldflags=-Wl,-z,lazy -s -w" \ github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk@${NVIDIA_CTK_VERSION} \ && cp ${GOPATH}/bin/nvidia-ctk /artifacts/nvidia-ctk -FROM nvcr.io/nvidia/cuda:12.6.1-base-ubuntu20.04 +FROM nvcr.io/nvidia/cuda:12.6.2-base-ubuntu20.04 # Remove CUDA libs(compat etc) in favor of libs installed by the NVIDIA driver RUN apt-get --purge -y autoremove cuda-*