reverting cuda/torch versions

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.0-base-ubuntu22.04
+FROM nvidia/cuda:11.8.0-base-ubuntu22.04
 
 ENV DEBIAN_FRONTEND=noninteractive \
     PYTHONUNBUFFERED=1 \
@@ -17,7 +17,7 @@ RUN --mount=type=cache,target=/var/cache/apt --mount=type=cache,target=/var/lib/
 
 RUN useradd -m -u 1000 ac
 RUN --mount=type=cache,target=/root/.cache python -m pip install --upgrade pip wheel
-ENV TORCH_COMMAND="pip install torch==2.1.0+cu121 torchaudio --extra-index-url https://download.pytorch.org/whl/cu121"
+ENV TORCH_COMMAND="pip install torch==2.0.1+cu118 torchaudio --extra-index-url https://download.pytorch.org/whl/cu118"
 RUN --mount=type=cache,target=/root/.cache python -m $TORCH_COMMAND
 RUN ln -s /usr/lib/x86_64-linux-gnu/libnvrtc.so.11.2 /usr/lib/x86_64-linux-gnu/libnvrtc.so
 USER 1000