diff --git a/gpu.Dockerfile b/gpu.Dockerfile
index 2132274..f2f6f51 100644
--- a/gpu.Dockerfile
+++ b/gpu.Dockerfile
@@ -4,7 +4,7 @@ WORKDIR /program
 
 COPY gpu-requirements.txt .
 
-RUN python3.9 -m pip install --no-cache-dir -r gpu-requirements.txt
+RUN pip3 install --no-cache-dir -r gpu-requirements.txt
 
 COPY / .
 
diff --git a/requirements.txt b/requirements.txt
index f54effb..867a911 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -320,12 +320,12 @@ tokenizers==0.21.2
     # via
     #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
     #   transformers
-torch==2.7.1+cpu
+torch==2.7.1
     # via
     #   -r /home/runner/work/refinery-submodule-parent-images/refinery-submodule-parent-images/refinery-embedder/requirements/torch-cpu-requirements.txt
     #   sentence-transformers
     #   torchvision
-torchvision==0.22.1+cpu
+torchvision==0.22.1
     # via -r requirements/requirements.in
 tqdm==4.67.1
     # via