[conda] fixed colossalai dependency version (#162)

* [conda] fixed colossalai dependency version * polish code

[conda] fixed colossalai dependency version (#162)
* [conda] fixed colossalai dependency version * polish code
867587b3 · Frank Lee · GitHub · a8b25e72 · 867587b3
Unverified Commit 867587b3 authored Mar 10, 2023 by Frank Lee Committed by GitHub Mar 10, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

environment.yml environment.yml +1 -2

No files found.
--- a/environment.yml
+++ b/environment.yml
@@ -17,8 +17,7 @@ dependencies:
      - ray==2.0.0
      - pyarrow
      - pandas
-      - --find-links https://release.colossalai.org
+      - colossalai==0.2.7
-      - colossalai==0.1.10+torch1.12cu11.3
  - pytorch::pytorch=1.12
  - pytorch::torchvision
  - pytorch::torchaudio