diff --git a/docker/gpu/Dockerfile b/docker/gpu/Dockerfile index ba6ea53..05716e6 100644 --- a/docker/gpu/Dockerfile +++ b/docker/gpu/Dockerfile @@ -1,4 +1,5 @@ #Updating CUDA to 12.4 to support older drivers and cards. +#testEdit FROM --platform=$BUILDPLATFORM nvidia/cuda:12.4.1-cudnn-runtime-ubuntu22.04 # Set non-interactive frontend ENV DEBIAN_FRONTEND=noninteractive