【学习】电脑上有多个GPU,命令行指定GPU进行训练。

使用如下指令可以指定使用的GPU。

CUDA_VISIBLE_DEVICES=1 

假设要使用第二个GPU进行训练。

CUDA_VISIBLE_DEVICES=1 python train.py

你可能感兴趣的:(学习,人工智能,深度学习)