import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

pytorch和TF都適合

config = tf.ConfigProto(allow_soft_placement=True)
config.gpu_options.per_process_gpu_memory_fraction =1  # 占用40%顯存
# sess = tf.Session(config=config)# with tf.Session() as sess:
with tf.Session(config=config) as sess:

with tf.Session() as sess:with tf.device("/gpu:0"):/gpu改成cpu

總結

以上是生活随笔為你收集整理的禁用gpu首选的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。