Update Dockerfile

Dockerfile

@@ -1,8 +1,13 @@
 FROM nvidia/cuda:12.1.0-base-ubuntu22.04
 
 USER root
-RUN apt-get update && apt-get install -y curl bash && \
-    mkdir -p /workspace/mount && \
+RUN apt-get update && apt-get install -y \
+    curl \
+    bash \
+    libatomic1 \
+    && rm -rf /var/lib/apt/lists/*
+
+RUN mkdir -p /workspace/mount && \
     chmod -R 777 /workspace && \
     useradd -m -u 1000 user