summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--environment.yaml6
1 files changed, 4 insertions, 2 deletions
diff --git a/environment.yaml b/environment.yaml
index 9af40eb..03345c6 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -1,15 +1,17 @@
1name: ldd 1name: ldd
2channels: 2channels:
3 - pytorch 3 - pytorch
4 - nvidia
4 - xformers/label/dev 5 - xformers/label/dev
5 - defaults 6 - defaults
6dependencies: 7dependencies:
7 - cudatoolkit=11.3 8 - cudatoolkit=11.3
9 - libcufile=1.4.0.31
8 - matplotlib=3.6.2 10 - matplotlib=3.6.2
9 - numpy=1.23.4 11 - numpy=1.23.4
10 - pip=22.3.1 12 - pip=22.3.1
11 - python=3.9.15 13 - python=3.10.8
12 - pytorch=1.13.1 14 - pytorch=1.13.1=*cuda*
13 - torchvision=0.14.1 15 - torchvision=0.14.1
14 - pip: 16 - pip:
15 - -e . 17 - -e .