diff --git a/environment.yml b/environment.yml
index d5cfcf8..186069e 100644
--- 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