newturok commited on
Commit
c11821c
1 Parent(s): 4d7bcfb

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -2
Dockerfile CHANGED
@@ -2,7 +2,7 @@ FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
2
  ENV DEBIAN_FRONTEND=noninteractive
3
  ENV TZ=Europe/Paris
4
  RUN apt-get update
5
- RUN apt-get install -y git git-lfs wget curl make cmake ffmpeg llvm python3
6
  RUN apt-get install -y build-essential xz-utils tk-dev libsm6 libxext6 libgl1-mesa-glx
7
  RUN apt-get install -y libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev libncursesw5-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev
8
  RUN rm -rf /var/lib/apt/lists/*
@@ -15,7 +15,6 @@ ENV HOME=/home/user
15
  ENV PATH=/home/user/.local/bin:$PATH
16
  WORKDIR /home/user/app
17
 
18
- RUN apt-get install python3-pip
19
  RUN pip install --no-cache-dir addict albumentations addict albumentations basicsr blendmodes einops gradio imageio imageio-ffmpeg invisible-watermark kornia numba omegaconf open_clip_torch prettytable pytorch-lightning safetensors streamlit streamlit-drawable-canvas test-tube timm torchmetrics transformers webdataset yapf
20
 
21
  ENV PYTHONPATH=$HOME/app
 
2
  ENV DEBIAN_FRONTEND=noninteractive
3
  ENV TZ=Europe/Paris
4
  RUN apt-get update
5
+ RUN apt-get install -y git git-lfs wget curl make cmake ffmpeg llvm python3 python3-pip
6
  RUN apt-get install -y build-essential xz-utils tk-dev libsm6 libxext6 libgl1-mesa-glx
7
  RUN apt-get install -y libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev libncursesw5-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev
8
  RUN rm -rf /var/lib/apt/lists/*
 
15
  ENV PATH=/home/user/.local/bin:$PATH
16
  WORKDIR /home/user/app
17
 
 
18
  RUN pip install --no-cache-dir addict albumentations addict albumentations basicsr blendmodes einops gradio imageio imageio-ffmpeg invisible-watermark kornia numba omegaconf open_clip_torch prettytable pytorch-lightning safetensors streamlit streamlit-drawable-canvas test-tube timm torchmetrics transformers webdataset yapf
19
 
20
  ENV PYTHONPATH=$HOME/app