cuda-device-ipython-magic A ipython-magic for set CUDA_VISIBLE_DEVCIES in ipython Usage In []: %load_ext cuda_device In []: %CUDA_VISIBLE_DEVICES 0,1 Installation pip install cuda-device-ipython-magic