diff options
Diffstat (limited to 'environment.yaml')
| -rw-r--r-- | environment.yaml | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/environment.yaml b/environment.yaml index 5ecc5a8..de35645 100644 --- a/environment.yaml +++ b/environment.yaml | |||
| @@ -6,7 +6,7 @@ dependencies: | |||
| 6 | - cudatoolkit=11.3 | 6 | - cudatoolkit=11.3 |
| 7 | - numpy=1.22.3 | 7 | - numpy=1.22.3 |
| 8 | - pip=20.3 | 8 | - pip=20.3 |
| 9 | - python=3.8.10 | 9 | - python=3.9.13 |
| 10 | - pytorch=1.12.1 | 10 | - pytorch=1.12.1 |
| 11 | - torchvision=0.13.1 | 11 | - torchvision=0.13.1 |
| 12 | - pandas=1.4.3 | 12 | - pandas=1.4.3 |
| @@ -32,6 +32,6 @@ dependencies: | |||
| 32 | - test-tube>=0.7.5 | 32 | - test-tube>=0.7.5 |
| 33 | - torch-fidelity==0.3.0 | 33 | - torch-fidelity==0.3.0 |
| 34 | - torchmetrics==0.9.3 | 34 | - torchmetrics==0.9.3 |
| 35 | - transformers==4.22.2 | 35 | - transformers==4.23.1 |
| 36 | - triton==2.0.0.dev20220924 | 36 | - triton==2.0.0.dev20220924 |
| 37 | - xformers==0.0.13 | 37 | - xformers==0.0.13 |