Commit 0595878b authored by duanjinfei's avatar duanjinfei

update cog.yaml

parent 2442868d
...@@ -9,13 +9,13 @@ build: ...@@ -9,13 +9,13 @@ build:
system_packages: system_packages:
- pip=22.2.1 - pip=22.2.1
- cudatoolkit=11.7 - cudatoolkit=11.7
- 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=1.23.3
- cython==0.29.33 - cython==0.29.33
- pytorch - pytorch
- defaults - defaults
......
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