diff options
Diffstat (limited to 'environment.yaml')
| -rw-r--r-- | environment.yaml | 6 |
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 @@ | |||
| 1 | name: ldd | 1 | name: ldd |
| 2 | channels: | 2 | channels: |
| 3 | - pytorch | 3 | - pytorch |
| 4 | - nvidia | ||
| 4 | - xformers/label/dev | 5 | - xformers/label/dev |
| 5 | - defaults | 6 | - defaults |
| 6 | dependencies: | 7 | dependencies: |
| 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 . |