Commit 2442868d authored by duanjinfei's avatar duanjinfei

update cog.yaml

parent 5adfd23d
...@@ -8,14 +8,13 @@ build: ...@@ -8,14 +8,13 @@ build:
# a list of ubuntu apt packages to install # a list of ubuntu apt packages to install
system_packages: system_packages:
- pip=22.2.1 - pip=22.2.1
- cudatoolkit=11.7
# 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:
- cudatoolkit=11.7
- numpy=1.23.3 - numpy=1.23.3
- cython==0.29.33 - cython==0.29.33
- pytorch - pytorch
......
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