leoxing1996 commited on
Commit
bd4d838
·
1 Parent(s): ca8f50a

update pip install cmd

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -2
Dockerfile CHANGED
@@ -46,9 +46,10 @@ COPY --chown=user . $HOME/app
46
  ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
47
 
48
  # install dependencies
49
- RUN pip install torch torchvision xformers --index-url https://download.pytorch.org/whl/cu121
 
50
  RUN git submodule update --init --recursive
51
- RUN pip install -e ."[tensorrt]"
52
 
53
  # download models`
54
  RUN mkdir models
 
46
  ENV LD_PRELOAD=/usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
47
 
48
  # install dependencies
49
+ RUN pip3 install --no-cache-dir --upgrade --pre -r /code/requirements.txt
50
+ RUN pip3 install torch torchvision xformers --index-url https://download.pytorch.org/whl/cu121
51
  RUN git submodule update --init --recursive
52
+ RUN pip3 install -e ."[tensorrt]"
53
 
54
  # download models`
55
  RUN mkdir models