Commit 360d8336 authored by duanjinfei's avatar duanjinfei

update cog.yaml

parent 69d5ea77
...@@ -10,9 +10,6 @@ build: ...@@ -10,9 +10,6 @@ build:
- wget - wget
- git - git
- curl - curl
- pytorch
- pip=22.2.1
- cudatoolkit
- libgl1-mesa-glx - libgl1-mesa-glx
- libglib2.0-0 - libglib2.0-0
...@@ -61,8 +58,7 @@ build: ...@@ -61,8 +58,7 @@ build:
# commands run after the environment is setup # commands run after the environment is setup
run: run:
- pip install Pillow==9.5.0 - pip install torch==2.0.1 torchvision==0.15.2
- pip install gradio==3.50.0
- "echo env is ready!" - "echo env is ready!"
- "echo another command if needed" - "echo another command if needed"
......
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