Spaces:
Configuration error
Configuration error
kzielins commited on
Commit ·
a8dd03e
1
Parent(s): d4ab283
conda requirements.txt
Browse files- environment.yml +8 -1
- requirements.txt +12 -0
environment.yml
CHANGED
|
@@ -1,7 +1,14 @@
|
|
| 1 |
name: motionbert
|
| 2 |
channels:
|
| 3 |
-
- conda-forge
|
|
|
|
| 4 |
- defaults
|
|
|
|
| 5 |
dependencies:
|
| 6 |
- python=3.10
|
| 7 |
- gradio
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
name: motionbert
|
| 2 |
channels:
|
| 3 |
+
- conda-forge
|
| 4 |
+
- pytorch
|
| 5 |
- defaults
|
| 6 |
+
- nvidia
|
| 7 |
dependencies:
|
| 8 |
- python=3.10
|
| 9 |
- gradio
|
| 10 |
+
- pytorch
|
| 11 |
+
- torchvision
|
| 12 |
+
- torchaudio
|
| 13 |
+
- pytorch-cuda=11.6
|
| 14 |
+
#conda install pytorch torchvision torchaudio pytorch-cuda=11.6 -c pytorch -c nvidia
|
requirements.txt
ADDED
|
@@ -0,0 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
tensorboardX
|
| 2 |
+
tqdm
|
| 3 |
+
easydict
|
| 4 |
+
prettytable
|
| 5 |
+
chumpy
|
| 6 |
+
opencv-python
|
| 7 |
+
imageio-ffmpeg
|
| 8 |
+
matplotlib==3.1.1
|
| 9 |
+
roma
|
| 10 |
+
ipdb
|
| 11 |
+
pytorch-metric-learning # For one-shot action recognition
|
| 12 |
+
smplx[all] # For mesh recovery
|