diff --git a/Dockerfile.dependencies b/Dockerfile.dependencies
index 9af6146..362cd17 100644
--- a/Dockerfile.dependencies
+++ b/Dockerfile.dependencies
@@ -1,4 +1,5 @@
-FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04
+# FROM nvidia/cuda:11.7.1-cudnn8-runtime-ubuntu22.04
+FROM ubuntu:22.04
 
 ARG PKGTMPDIR=/tmp/pkgs
 
@@ -242,7 +243,7 @@ RUN mkdir -p ${PKGTMPDIR} && cd ${PKGTMPDIR} \
     && python -m pip install cmake \
     && python -m pip install numpy \
     && python -m pip install jep==4.0.3 \
-    && python -m pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 \
+    # && python -m pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117 \
     && echo "Due to container high storage usage, Tensorflow was removed - it's only used on yahoo image detection - if you want to use it, uncomment the following line:" \
     && echo "python -m pip install tensorflow==2.9.1 tensorrt==8.5.3.1" \
     && python -m pip install keras \