Commit 55818bfa authored by duanjinfei's avatar duanjinfei

update cog.yaml

parent 62590c76
......@@ -7,18 +7,19 @@ build:
# a list of ubuntu apt packages to install
system_packages:
- pytorch
- defaults
- pip=23.0.1
- cudatoolkit=11.7
- numpy=1.23.3
- cython==0.29.33
- pip=24.1.1
# python version in the form '3.11' or '3.11.4'
python_version: "3.10.6"
# a list of packages in the format <package-name>==<version>
python_packages:
- cudatoolkit=11.7
- numpy=1.23.3
- cython==0.29.33
- pytorch
- defaults
- Pillow==9.5.0
- gradio==3.50.0
- albumentations==0.4.3
......
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