GitHub - lanpa/CUDA_VISIBLE_DEVICES-tool: convenient script to switch visible Nvidia GPU. (original) (raw)

usegpu

A convenient script to switch visible Nvidia GPU.

install

wget https://raw.githubusercontent.com/lanpa/usegpu/master/appendtobash && cat appendtobash>>~/.bashrc source ~/.bashrc

usage

usegpu 1 # use gpu 1 usegpu 1,2 #usegpu 1 and 2 usegpu 0,2 #usegpu 0 and 2 usegpu #show current visible gpu