update cog.yaml

69d5ea77 · duanjinfei · 6b083396 · 69d5ea77
Commit 69d5ea77 authored Jul 16, 2024 by duanjinfei
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

cog.yaml cog.yaml +6 -0

No files found.
--- a/cog.yaml
+++ b/cog.yaml
@@ -10,6 +10,12 @@ build:
    - wget
    - git
    - curl
+    - pytorch
+    - pip=22.2.1
+    - cudatoolkit
+    - libgl1-mesa-glx
+    - libglib2.0-0
+

  # python version in the form '3.11' or '3.11.4'
  python_version: "3.10.6"