Skip to content

Commit

Permalink
Update Dockerfile
Browse files Browse the repository at this point in the history
  • Loading branch information
cssprad1 authored Nov 21, 2024
1 parent 1a80649 commit b587c34
Showing 1 changed file with 2 additions and 6 deletions.
8 changes: 2 additions & 6 deletions requirements/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@ FROM ${FROM_IMAGE}:${VERSION_DATE}-py3

# Ubuntu needs noninteractive to be forced
ENV DEBIAN_FRONTEND noninteractive
ENV PROJ_LIB="/usr/share/proj"
ENV CPLUS_INCLUDE_PATH="/usr/include/gdal"
ENV C_INCLUDE_PATH="/usr/include/gdal"

Expand Down Expand Up @@ -75,13 +74,13 @@ RUN git clone --single-branch --branch master https://github.com/pkolano/shift.g
rm -rf /app

# Pip
RUN pip --no-cache-dir install omegaconf \
RUN pip --no-cache-dir install \
pytorch-lightning \
Lightning \
transformers \
datasets \
webdataset \
deepspeed \
webdataset \
'huggingface_hub[cli,torch]' \
torchgeo \
rasterio \
Expand All @@ -95,9 +94,6 @@ RUN pip --no-cache-dir install omegaconf \
opencv-contrib-python-headless \
tifffile \
webcolors \
Pillow \
seaborn \
xgboost \
tiler \
segmentation-models \
timm \
Expand Down

0 comments on commit b587c34

Please sign in to comment.