TensorFlow使用GPU

1.在運行以前先查看GPU的使用狀況:  指令:nvidia-smi 備註:查看GPU此時的使用狀況python 2.指定GPU訓練:  方法一、在python程序中設置:  代碼:os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘0’ 備註:使用 GPU 0  代碼:os.environ[‘CUDA_VISIBLE_DEVICES’] = ‘0,1’ 備註:使用 GPU
相關文章
相關標籤/搜索