Commit 2442868d authored by duanjinfei's avatar duanjinfei

update cog.yaml

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