我可以在tensorflow 2.x中将一个物理GPU拆分为两个虚拟设备吗?

我可以在tensorflow 2.x中将一个物理GPU拆分为两个虚拟设备吗?,tensorflow,gpu,nvidia,tensorflow2.x,Tensorflow,Gpu,Nvidia,Tensorflow2.x,我想测试使用多个GPU来训练网络。但是,我的机器上只安装了一个物理GPU(RTX 2070)。我能否将此设备拆分为两个虚拟设备? 我当前的尝试是基于tf.config.experimental.set\u virtual\u device\u configuration函数,但是,它似乎不起作用。完整示例: 将tensorflow导入为tf 打印(tf.uuu版本)#2.4.1 物理\GPU=tf.config.experimental.list\u物理\u设备(“GPU”) 打印(物理GPU)

我想测试使用多个GPU来训练网络。但是,我的机器上只安装了一个物理GPU(RTX 2070)。我能否将此设备拆分为两个虚拟设备?
我当前的尝试是基于
tf.config.experimental.set\u virtual\u device\u configuration
函数,但是,它似乎不起作用。完整示例:

将tensorflow导入为tf
打印(tf.uuu版本)#2.4.1
物理\GPU=tf.config.experimental.list\u物理\u设备(“GPU”)
打印(物理GPU)#[物理设备(名称='/物理设备:GPU:0',设备类型='GPU')]
tf.config.experimental.set\u虚拟设备\u配置(
物理图形处理器[0],
[tf.config.experimental.VirtualDeviceConfiguration(内存限制=2048),
tf.config.experimental.VirtualDeviceConfiguration(内存限制=2048)]
)
打印(tf.config.experimental.list_physical_devices(“GPU”))35;[PhysicalDevice(name='/physical_device:GPU:0',device_type='GPU')]

应使用
列表\u逻辑\u设备验证可见的GPU设备总数

logical\u GPU=tf.config.experimental.list\u逻辑\u设备('GPU'))
打印(逻辑处理器)
#[逻辑设备(名称='/设备:GPU:0',设备类型='GPU'),
#逻辑设备(名称='/device:GPU:1',device_type='GPU')]