前言

有時候因應我們有訓練時間的需求，
我們會在一台主機上插上多張 GPU，
我們可以使用以下方式，
在同一台電腦上指定 GPU 進行運算。

• (如果沒有指定，預設都是 第0張 GPU)

Sample Code

預設：只使用第一張 GPU (GPU0)

指定用第二張 GPU (GPU1) 運行程式

CUDA_VISIBLE_DEVICES=1 python train.py

同時使用多張 GPU 運行程式

CUDA_VISIBLE_DEVICES=0,1,2,3 python train.py

Reference

https://stackoverflow.com/questions/40726039/tensorflow-cuda-visible-devices-doesnt-seem-to-work