diff --git a/geo-notebook/cuda12-pytorch-2.2.2/Dockerfile b/geo-notebook/cuda12-pytorch-2.2.2/Dockerfile
index 5408efc..12ac865 100644
--- a/geo-notebook/cuda12-pytorch-2.2.2/Dockerfile
+++ b/geo-notebook/cuda12-pytorch-2.2.2/Dockerfile
@@ -24,7 +24,8 @@ USER root
 # ImportError: /opt/conda/lib/python3.11/lib-dynload/_sqlite3.cpython-311-x86_64-linux-gnu.so: undefined symbol: sqlite3_deserialize
 # https://github.com/proxystore/taps/issues/151#issuecomment-2339537161
 RUN mamba install -y libsqlite==3.45.* && \
-    mamba install -y -c conda-forge segment-geospatial
+    mamba install -c conda-forge segment-geospatial "pytorch=*=cuda*" && \
+    mamba install -c conda-forge groundingdino-py segment-anything-fast
 
 RUN rm /tmp/requirements.txt && \
     /opt/conda/bin/conda clean -afy && \