Colossal-AI-readme-7.sh
· 307 B · Bash
原始檔案
# clone the repository
git clone https://github.com/hpcaitech/ColossalAI.git
cd ColossalAI
# download the cub library
wget https://github.com/NVIDIA/cub/archive/refs/tags/1.8.0.zip
unzip 1.8.0.zip
cp -r cub-1.8.0/cub/ colossalai/kernel/cuda_native/csrc/kernels/include/
# install
BUILD_EXT=1 pip install .
| 1 | # clone the repository |
| 2 | git clone https://github.com/hpcaitech/ColossalAI.git |
| 3 | cd ColossalAI |
| 4 | |
| 5 | # download the cub library |
| 6 | wget https://github.com/NVIDIA/cub/archive/refs/tags/1.8.0.zip |
| 7 | unzip 1.8.0.zip |
| 8 | cp -r cub-1.8.0/cub/ colossalai/kernel/cuda_native/csrc/kernels/include/ |
| 9 | |
| 10 | # install |
| 11 | BUILD_EXT=1 pip install . |