Tensorflow 2.0 GPU的使用與限制使用率及虛擬多GPU

先插入一行簡單代碼,如下複製便可用來設置GPU使用率:python import tensorflow as tf import numpy as np print(tf.__version__) import os # 設置可以使用的 gpu 序號 os.environ['CUDA_VISIBLE_DEVICES'] = '0' # 用來設置是否在特殊狀況下在cpu上進行計算 tf.conf
相關文章
相關標籤/搜索