Updated the dependencies.

Updated the dependencies to follow instructions on https://pytorch.org/get-started/locally/
LyaaaaaGames · Apr 10, 2023 · 6f0821e · 6f0821e
1 parent d7042be
commit 6f0821e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/conda_config.yml b/conda_config.yml
@@ -9,4 +9,7 @@ dependencies:
   - websockets=10.4
   - transformers=4.27
   - sentencepiece
-  - accelerate
+  - accelerate
+  - torchvision
+  - torchaudio
+  - cpuonly
diff --git a/conda_config_cuda.yml b/conda_config_cuda.yml
@@ -2,13 +2,15 @@ name: aidventure
 channels:
   - conda-forge
   - pytorch
+  - nvidia
 dependencies:
   - pip
   - python=3.9.7
   - pytorch
+  - torchvision
+  - torchaudio
+  - pytorch-cuda=11.8
   - websockets=10.4
   - transformers=4.27
   - sentencepiece
-  - accelerate
-  - torchvision
-  - cudatoolkit
+  - accelerate