Merge 0a781bf14a into ef1ac08ce3

2025-04-05 02:20:47 +00:00 · 2023-09-20 19:19:59 +08:00 · 2023-09-20 19:19:59 +08:00 · 9c689a6da7
commit 9c689a6da7
parent ef1ac08ce3 0a781bf14a
1 changed files with 4 additions and 3 deletions
--- a/environment.yml
+++ b/environment.yml
@ -3,14 +3,15 @@ channels:
  - pytorch
  - defaults
  - anaconda
+  - nvidia
 dependencies:
  - python=3.9
  - cudatoolkit
  - pip
-  - pytorch=1.12.1
+  - pytorch=1.13.1
  - pytorch-mutex=1.0=cuda
-  - torchaudio=0.12.1
-  - torchvision=0.13.1
+  - torchaudio=0.13.1
+  - torchvision=0.14.1
  - pip:
    - accelerate==0.16.0
    - aiohttp==3.8.4