environment file

osanseviero · Feb 11, 2023 · ddccacd · ddccacd
1 parent 67963c9
commit ddccacd
Showing 1 changed file with 23 additions and 0 deletions.
diff --git a/environment.yml b/environment.yml
@@ -0,0 +1,23 @@
+name: pix2pix-zero
+channels:
+  - pytorch
+  - nvidia
+  - defaults
+dependencies:
+  - pip
+  - pytorch-cuda=11.6
+  - torchvision
+  - pytorch
+  - pip:
+    - accelerate
+    - diffusers
+    - einops
+    - gradio
+    - ipython
+    - numpy
+    - opencv-python-headless
+    - pillow
+    - psutil
+    - tqdm
+    - transformers
+    - salesforce-lavis