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 @@
 name: ldd
 channels:
   - pytorch
+  - nvidia
   - xformers/label/dev
   - defaults
 dependencies:
   - cudatoolkit=11.3
+  - libcufile=1.4.0.31
   - matplotlib=3.6.2
   - numpy=1.23.4
   - pip=22.3.1
-  - python=3.9.15
-  - pytorch=1.13.1
+  - python=3.10.8
+  - pytorch=1.13.1=*cuda*
   - torchvision=0.14.1
   - pip:
       - -e .