Realcat commited on
Commit
16a23bd
1 Parent(s): 5fdb334

update: env

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. requirements.txt +4 -4
Dockerfile CHANGED
@@ -1,5 +1,5 @@
1
  # Use an official conda-based Python image as a parent image
2
- FROM pytorch/pytorch:2.2.1-cuda12.1-cudnn8-runtime
3
  LABEL maintainer vincentqyw
4
  ARG PYTHON_VERSION=3.10.10
5
 
 
1
  # Use an official conda-based Python image as a parent image
2
+ FROM pytorch/pytorch:2.4.0-cuda12.1-cudnn9-runtime
3
  LABEL maintainer vincentqyw
4
  ARG PYTHON_VERSION=3.10.10
5
 
requirements.txt CHANGED
@@ -2,8 +2,8 @@ e2cnn
2
  einops
3
  easydict
4
  gdown
5
- gradio==4.28.3
6
- gradio_client==0.16.0
7
  h5py
8
  huggingface_hub
9
  imageio
@@ -21,7 +21,7 @@ psutil
21
  plotly
22
  protobuf
23
  poselib
24
- pycolmap
25
  pytlsd
26
  PyYAML
27
  pytorch-lightning==1.4.9
@@ -32,7 +32,7 @@ seaborn
32
  shapely
33
  tensorboardX==2.6.1
34
  torchmetrics==0.6.0
35
- torchvision==0.17.1
36
  roma #dust3r
37
  tqdm
38
  yacs
 
2
  einops
3
  easydict
4
  gdown
5
+ gradio==4.44.0
6
+ gradio_client==1.3.0
7
  h5py
8
  huggingface_hub
9
  imageio
 
21
  plotly
22
  protobuf
23
  poselib
24
+ pycolmap==0.6.1
25
  pytlsd
26
  PyYAML
27
  pytorch-lightning==1.4.9
 
32
  shapely
33
  tensorboardX==2.6.1
34
  torchmetrics==0.6.0
35
+ torchvision==0.19.0
36
  roma #dust3r
37
  tqdm
38
  yacs