请问一下怎么在cpu上运行？

[PeyLiang]

我在运行train.sh的时候，调用的脚本是：
python -m torch.distributed.launch --nproc_per_node 1 train_parallel.py
这个里面使用cuda运行的，而我没有英伟达的显卡，想用cpu跑，应该怎么修改python脚本呢？
先谢谢了。