Skip to content

Commit

Permalink
fix pytorch gpu not found and add groundingdino and samfast
Browse files Browse the repository at this point in the history
  • Loading branch information
mhwasil committed Nov 4, 2024
1 parent 2c4ed04 commit 2325ee3
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion geo-notebook/cuda12-pytorch-2.2.2/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -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 && \
Expand Down

0 comments on commit 2325ee3

Please sign in to comment.