Commit eeaa4dc8 authored by duanjinfei's avatar duanjinfei

update cog.yaml

parent 0595878b
...@@ -7,17 +7,17 @@ build: ...@@ -7,17 +7,17 @@ build:
# a list of ubuntu apt packages to install # a list of ubuntu apt packages to install
system_packages: system_packages:
- pytorch
- pip=22.2.1 - pip=22.2.1
- cudatoolkit=11.7 - cudatoolkit
- numpy=1.23.3
# python version in the form '3.11' or '3.11.4' # python version in the form '3.11' or '3.11.4'
python_version: "3.10.6" python_version: "3.10.6"
# a list of packages in the format <package-name>==<version> # a list of packages in the format <package-name>==<version>
python_packages: python_packages:
- numpy
- cython==0.29.33 - cython==0.29.33
- pytorch
- defaults - defaults
- Pillow==9.5.0 - Pillow==9.5.0
- gradio==3.50.0 - gradio==3.50.0
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment