diff --git a/Dockerfile b/Dockerfile
index 6e3e0cf..02f8c28 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -20,11 +20,11 @@ COPY data/ app/data/
 # Install requirements
 RUN conda install python=3.10
 RUN conda install -c conda-forge xesmf esmpy h5py -y
-RUN pip install torch --index-url https://download.pytorch.org/whl/cpu
+RUN pip install torch==2.2.0 --index-url https://download.pytorch.org/whl/cpu
 RUN pip install git+https://github.com/SheffieldSolar/PV_Live-API#pvlive_api
 
 # install cpu torch
-RUN pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
+RUN pip install torch==2.2.0 torchvision --index-url https://download.pytorch.org/whl/cpu
 
 # Change to app folder
 WORKDIR /app