Fix GPU dockerfile

gpu.Dockerfile

@@ -37,7 +37,7 @@ COPY --from=builder /code /code
 COPY --from=builder /usr/local /usr/local
 
 # Install OpenCL Drivers
-RUN apt install nvidia-opencl-icd-340 intel-opencl-icd
+RUN apt update && apt install -y nvidia-opencl-icd-340 intel-opencl-icd
 
 # Install shared libraries that we depend on via APT, but *not*
 # the -dev packages to save space!