Python指定GPU来跑代码：通俗易懂，亲测有用

1.在机子有多张显卡的情况下，可以使用 nvidia-smi查看所有显卡的编号及使用情况 ，如下：

2.可以看到，笔者的机子总共有两张显卡，分别是GPU0和GPU1。其中，显卡0相对来说更为空闲。

3.因此，笔者可以采用如下语句指定GPU0来跑代码：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

该语句一般放在main.py的最开始部分，大家根据自己的情况进行修改即可。