1. pip
pip安装包 | pip install -r requirements.txt | |
pip查看可用版本 | pip index versions numpy | |
pip升级 | pip install –upgrade pip | |
2. 运行
调试 | python -m debugpy –listen localhost:5678 –wait-for-client $(which torchrun) –nnodes=1 –nproc_per_node=1 train.py –config configs/training/v1/config.yaml |
指定GPU | CUDA_VISIBLE_DEVICES=gpu_ids test.py |