java tensorflow 并行session指定不同GPU
通过 DEVICE_LIST指定 gpu列表属于进程级别,在第二次初始化session指定不同gpu时将会报错。 GPUOptions gpus = GPUOptions.newBuilder().setVisibleDeviceList("2,3").setAllowGrowth(true).build(); ConfigProto conf = ConfigProto.newBuilder().setAllowSoftPlacement(false)
原创
2020-12-03 14:24:49 ·
462 阅读 ·
0 评论