Update Dockerfile-cuda

docker/Dockerfile-cuda

@@ -6,10 +6,12 @@ FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 # Arguments
 # ---------
 
+# Use ofed_info -s to get your local version
+ARG MOFED_VERSION=5.4-3.4.0.0
+ARG CONFIG=standard.yaml
 ARG ARCH=cuda
-ENV MILABENCH_GPU_ARCH=$ARCH
 
-ARG CONFIG=standard.yaml
+ENV MILABENCH_GPU_ARCH=$ARCH
 ENV MILABENCH_CONFIG_NAME=$CONFIG
 ENV MILABENCH_DOCKER=1
 
@@ -36,15 +38,12 @@ COPY . /milabench/milabench/
 #           rustc: used by BERT models inside https://pypi.org/project/tokenizers/
 # build-essential: for rust
 
-# Use ofed_info -s to get your local version
-ARG MOFED_VERSION=5.4-3.4.0.0
 
 ENV DEBIAN_FRONTEND=noninteractive
 RUN apt-get update -y &&\
-    apt-get install -y --no-install-recommends git build-essential gnupg2 curl ca-certificates python3 python-is-python3 python3-pip &&\
+    apt-get install -y --no-install-recommends git build-essential curl python3 python-is-python3 python3-pip &&\
     curl -o /etc/apt/trusted.gpg.d/mellanox.asc https://content.mellanox.com/ofed/RPM-GPG-KEY-Mellanox &&\
     curl -o /etc/apt/sources.list.d/mellanox.list https://linux.mellanox.com/public/repo/mlnx_ofed/${MOFED_VERSION}/ubuntu22.04/mellanox_mlnx_ofed.list &&\
-    dpkg -i cuda-keyring_1.1-1_all.deb &&\
     apt-get update -y &&\
     apt-get install -y --no-install-recommends libibverbs1 &&\
     apt-get clean &&\