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
相关文章
相关标签/搜索