服务器中指定GPU运行程序

1. 使用命令nvidia-smi查看有哪些可以使用的GPU

2 指定GPU运行Python程序

CUDA_VISIBLE_DEVICES=1 python main.py --cuda

你可能感兴趣的:(python,gpu)