Spaces:
Running
Running
Realcat
commited on
Commit
•
16a23bd
1
Parent(s):
5fdb334
update: env
Browse files- Dockerfile +1 -1
- 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.
|
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.
|
6 |
-
gradio_client==
|
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.
|
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
|