Son aktif 1728779694

knox revised this gist 1728779694. Go to revision

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 .
Newer Older