mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-03 18:19:58 +00:00
[doc] add os scope, update tutorial install and tips (#2914)
This commit is contained in:
@@ -1,5 +1,7 @@
|
||||
# Colossal-AI Tutorial Hands-on
|
||||
|
||||
> This path is an abbreviated tutorial prepared for specific activities and may not be maintained in real time. For use of Colossal-AI, please refer to other [examples](https://github.com/hpcaitech/ColossalAI/tree/main/examples) and [documents](https://www.colossalai.org/).
|
||||
|
||||
## Introduction
|
||||
|
||||
Welcome to the [Colossal-AI](https://github.com/hpcaitech/ColossalAI) tutorial, which has been accepted as official tutorials by top conference [SC](https://sc22.supercomputing.org/), [AAAI](https://aaai.org/Conferences/AAAI-23/), [PPoPP](https://ppopp23.sigplan.org/), [CVPR](https://cvpr2023.thecvf.com/), etc.
|
||||
@@ -38,16 +40,7 @@ If you encounter any problem while running these tutorials, you may want to rais
|
||||
|
||||
## 🛠️ Setup environment
|
||||
[[video]](https://www.youtube.com/watch?v=dpMYj974ZIc) You should use `conda` to create a virtual environment, we recommend **python 3.8**, e.g. `conda create -n colossal python=3.8`. This installation commands are for CUDA 11.3, if you have a different version of CUDA, please download PyTorch and Colossal-AI accordingly.
|
||||
|
||||
```
|
||||
# install torch
|
||||
# visit https://pytorch.org/get-started/locally/ to download other versions
|
||||
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
|
||||
|
||||
# install latest ColossalAI
|
||||
# visit https://colossalai.org/download to download corresponding version of Colossal-AI
|
||||
pip install colossalai==0.1.11rc3+torch1.12cu11.3 -f https://release.colossalai.org
|
||||
```
|
||||
You can refer to the [Installation](https://github.com/hpcaitech/ColossalAI#installation) to set up your environment.
|
||||
|
||||
You can run `colossalai check -i` to verify if you have correctly set up your environment 🕹️.
|
||||

|
||||
|
Reference in New Issue
Block a user