Merge pull request #11 from SamThe602/Ubuntu-24.04-cuda-13.1.1-minimal

Ubuntu 24.04 cuda 13.1.1 minimal

Author: JRPan

Dockerfile

@@ -1,6 +1,6 @@
-FROM nvidia/cuda:12.9.1-cudnn-devel-ubuntu24.04
+FROM nvidia/cuda:13.1.1-cudnn-devel-ubuntu24.04
 LABEL org.opencontainers.image.source=https://github.com/accel-sim/Dockerfile/
-LABEL org.opencontainers.image.description="Accel-Sim container with Ubuntu 24.04 with CUDA 12.8.0 and CUDNN, for CI runs only"
+LABEL org.opencontainers.image.description="Accel-Sim container with Ubuntu 24.04 with CUDA 13.1.1 and CUDNN, for CI runs only"
 
 SHELL ["/bin/bash", "-c"]
 
@@ -12,6 +12,7 @@ ENV BOOST_ROOT=/usr/include/boost
 ENV PATH=$CUDA_INSTALL_PATH/bin:$PATH
 
 ENV GPUAPPS_ROOT=/accel-sim/gpu-app-collection
+ARG GPU_APP_COLLECTION_BRANCH=dev
 
 RUN apt-get update && apt-get install -y wget build-essential xutils-dev bison zlib1g-dev flex \
       libglu1-mesa-dev git g++ libssl-dev libxml2-dev libboost-all-dev git g++ \
@@ -27,7 +28,8 @@ RUN pip3 install --upgrade pip
 RUN pip3 install pyyaml plotly psutil
 
 # For CI, only build the CI apps and pull the regression data
-# Clone the gpu-app-collection repository
-RUN git clone --recurse-submodules https://github.com/accel-sim/gpu-app-collection.git
+# Clone the gpu-app-collection repository. 
+RUN git clone --recurse-submodules --branch ${GPU_APP_COLLECTION_BRANCH} --single-branch \
+      https://github.com/accel-sim/gpu-app-collection.git
 # Build the CI apps and pull the regression data
-RUN cd gpu-app-collection && bash test-build.sh ci && bash get_regression_data.sh
+RUN cd gpu-app-collection && bash test-build.sh ci && bash get_regression_data.sh