set cuda visible devices python

The solution for this is noted below

set cuda visible devices python

Solution

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0"

Try other methods by searching on the site. That is if this doesn’t work