前言
有時候因應我們有訓練時間的需求,
我們會在一台主機上插上多張 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