import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
注意:上述代码一定要放在import tensorflow或keras等之前,否则不起作用。
参考文献:https://blog.csdn.net/bkk854762363/article/details/84236389
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
注意:上述代码一定要放在import tensorflow或keras等之前,否则不起作用。
参考文献:https://blog.csdn.net/bkk854762363/article/details/84236389