3v324v23 commited on
Commit
b9a79d3
1 Parent(s): 8154f8b
Files changed (2) hide show
  1. Dockerfile +0 -1
  2. requirements.txt +19 -0
Dockerfile CHANGED
@@ -30,7 +30,6 @@ RUN pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url h
30
 
31
  COPY . /home/${USER_NAME}
32
  WORKDIR /home/${USER_NAME}/
33
- RUN pip install -r requirements/minimal.txt -f https://download.pytorch.org/whl/cu118/torch_stable.html
34
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
35
 
36
 
 
30
 
31
  COPY . /home/${USER_NAME}
32
  WORKDIR /home/${USER_NAME}/
 
33
  RUN pip install --no-cache-dir --upgrade -r requirements.txt
34
 
35
 
requirements.txt CHANGED
@@ -1,3 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  einops
2
  imageio
3
  gradio
 
1
+ tensorboard==2.7.0
2
+ torch-tb-profiler==0.3.1
3
+ ninja==1.10.2
4
+ numpy
5
+ scipy==1.7.3
6
+ scikit-learn==1.0.2
7
+ scikit-video==1.1.11
8
+ pillow==9.0.0
9
+ opencv-python-headless==4.5.5.62
10
+ requests
11
+ bs4
12
+ tqdm
13
+ rich
14
+ click
15
+ cloup
16
+ psutil
17
+ easydict
18
+ lmdb
19
+ matplotlib
20
  einops
21
  imageio
22
  gradio