Update cuda.yml

painebenjamin · Jan 18, 2024 · 374e9d4 · 374e9d4
1 parent 78108f4
commit 374e9d4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/environments/build/ubuntu-latest/cuda.yml b/environments/build/ubuntu-latest/cuda.yml
@@ -12,7 +12,7 @@ dependencies:
   - make=4.3
   - nodejs
   - ffmpeg
-  - opencv
+  - opencv=4.8.1
   - openh264
   - git
   - pip:
@@ -71,6 +71,8 @@ dependencies:
     - xformers>=0.0.20
     - open_clip_torch>=2.12.0
     - pyinstaller-hooks-contrib==2023.11
+    - opencv-python==4.8.1.78
+    - opencv-python-headless==4.8.1.78
     - git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers
     - git+https://github.com/openai/CLIP.git@main#egg=clip
     - git+https://github.com/CompVis/stable-diffusion.git@main#egg=latent-diffusion