mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-27 19:36:13 +00:00
68359ed1e1
* [release] update version * [devops] update compatibility test * [devops] update compatibility test * [devops] update compatibility test * [devops] update compatibility test * [test] fix ddp plugin test * [test] fix gptj and rpc test * [devops] fix cuda ext compatibility * [inference] fix flash decoding test * [inference] fix flash decoding test
13 lines
422 B
JSON
13 lines
422 B
JSON
{
|
|
"build": [
|
|
{
|
|
"torch_command": "pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu121",
|
|
"cuda_image": "hpcaitech/cuda-conda:12.1"
|
|
},
|
|
{
|
|
"torch_command": "pip install torch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 --index-url https://download.pytorch.org/whl/cu118",
|
|
"cuda_image": "hpcaitech/cuda-conda:11.8"
|
|
}
|
|
]
|
|
}
|