From 8a06ba802b742de419f0abae6d13523a7ebc3f83 Mon Sep 17 00:00:00 2001 From: makaveli10 Date: Mon, 4 Mar 2024 03:24:49 -0500 Subject: [PATCH] update cuda version 12.2.2 gpu dockerfile --- docker/Dockerfile.gpu | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile.gpu b/docker/Dockerfile.gpu index b3cc44e..6775b89 100644 --- a/docker/Dockerfile.gpu +++ b/docker/Dockerfile.gpu @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 +FROM nvidia/cuda:12.2.2-cudnn8-runtime-ubuntu22.04 ARG DEBIAN_FRONTEND=noninteractive # Remove any third-party apt sources to avoid issues with expiring keys.