Skip to content

GitLab

You need to sign in or sign up before continuing.
Unverified Commit 867587b3 authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[conda] fixed colossalai dependency version (#162) 

* [conda] fixed colossalai dependency version

* polish code
parent a8b25e72
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
environment.yml
View file @ 867587b3
......@@ -17,8 +17,7 @@ dependencies:
- ray==2.0.0
- pyarrow
- pandas
- --find-links https://release.colossalai.org
- colossalai==0.1.10+torch1.12cu11.3
- colossalai==0.2.7
- pytorch::pytorch=1.12
- pytorch::torchvision
- pytorch::torchaudio
......
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