github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-31 16:40:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

[workflow] supported test on CUDA 10.2 (#3841)

Browse Source
This commit is contained in:
Frank Lee
2023-05-25 14:14:34 +08:00
committed by GitHub
parent 3229f93e30
commit 54e97ed7ea
5 changed files with 90 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
docs/source/en/get_started/installation.md
View File

@@ -48,5 +48,20 @@ If you don't want to install and enable CUDA kernel fusion (compulsory installat
pip install .
```
For Users with CUDA 10.2, you can still build ColossalAI from source. However, you need to manually download the cub library and copy it to the corresponding directory.
```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
CUDA_EXT=1 pip install .
```
<!-- doc-test-command: echo "installation.md does not need test" -->

16
docs/source/zh-Hans/get_started/installation.md
View File

@@ -47,4 +47,20 @@ CUDA_EXT=1 pip install .
pip install .
```
如果您在使用CUDA 10.2，您仍然可以从源码安装ColossalA。但是您需要手动下载cub库并将其复制到相应的目录。
```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
CUDA_EXT=1 pip install .
```
<!-- doc-test-command: echo "installation.md does not need test" -->