knox 修订了这个 Gist . 跳至此修订
1 file changed, 11 insertions
Colossal-AI-readme-7.sh(file created)
| @@ -0,0 +1,11 @@ | |||
| 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 . | |
更新
更早