8 lines
268 B
Bash
8 lines
268 B
Bash
|
export CUDA_VISIBLE_DEVICES=0,1,2,3
|
||
|
|
export NGPUS=4
|
||
|
|
export OMP_NUM_THREADS=1 # you can change this value according to your number of cpu cores
|
||
|
|
|
||
|
|
|
||
|
|
python -m torch.distributed.launch --nproc_per_node=$NGPUS train.py configs/culane.py
|
||
|
|
# python train.py configs/tusimple.py
|