Missing cmake

Dockerfile

@@ -1,4 +1,4 @@
-ARG UBUNTU_VERSION=20.04
+ARG UBUNTU_VERSION=22.04
 ARG CUDA_VERSION=12.3.1
 ARG BASE_CUDA_DEV_CONTAINER=nvidia/cuda:${CUDA_VERSION}-devel-ubuntu${UBUNTU_VERSION}
 ARG BASE_CUDA_RUN_CONTAINER=nvidia/cuda:${CUDA_VERSION}-runtime-ubuntu${UBUNTU_VERSION}
@@ -8,7 +8,7 @@ FROM ${BASE_CUDA_DEV_CONTAINER} as build
 ARG CUDA_DOCKER_ARCH=all
 
 RUN apt-get update && apt-get upgrade -y && \
-    apt-get install -y git build-essential gcc wget
+    apt-get install -y git build-essential gcc wget cmake
 
 WORKDIR /build
 
@@ -25,6 +25,8 @@ RUN mkdir build && \
     cmake --build . --config Release
 
 FROM ${BASE_CUDA_RUN_CONTAINER} as runtime
+RUN apt-get update && apt-get upgrade -y && \
+    apt-get install -y wget
 
 WORKDIR /app