watch -n 10 nvidia-smi
TensorFlow指定GPU
CUDA_DEVICE_ORDER="PCI_BUS_ID" CUDA_VISIBLE_DEVICES="0"
Theano指定GPU
THEANO_FLAGS=\'floatX=float32,device=gpu1\'
watch -n 10 nvidia-smi
TensorFlow指定GPU
CUDA_DEVICE_ORDER="PCI_BUS_ID" CUDA_VISIBLE_DEVICES="0"
Theano指定GPU
THEANO_FLAGS=\'floatX=float32,device=gpu1\'
相关文章: