mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-04-27 11:31:58 +00:00
17 lines
683 B
JSON
17 lines
683 B
JSON
{
|
|
"build": [
|
|
{
|
|
"torch_command": "pip install torch==1.12.1+cu102 torchvision==0.13.1+cu102 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu102",
|
|
"cuda_image": "hpcaitech/cuda-conda:10.2"
|
|
},
|
|
{
|
|
"torch_command": "pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113",
|
|
"cuda_image": "hpcaitech/cuda-conda:11.3"
|
|
},
|
|
{
|
|
"torch_command": "pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116",
|
|
"cuda_image": "hpcaitech/cuda-conda:11.6"
|
|
}
|
|
]
|
|
}
|