ubuntu_指定 CPU 或 GPU 运行代码

指定 GPU 有两种方式：

1. 在终端中运行：
   打开终端；
   输入命令：

export CUDA_VISIBLE_DEVICES=2

在当前终端里指定 GPU_id 为 2 的 gpu

2. 在代码中指定：

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

指定第 1 块 GPU 运行

3. 指定 CPU 运行：

import os
os.environ["export CUDA_VISIBLE_DEVICES"] = '-1'

指定 CPU 运行

4. 查看 gpu 资源：

watch -n 0.5 nvidia-smi

每 0.5 刷新一次 gpu 使用状况