You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: rfcs/20200624-pluggable-device-for-tensorflow.md
+8-8Lines changed: 8 additions & 8 deletions
Original file line number
Diff line number
Diff line change
@@ -75,7 +75,7 @@ This section describes the user scenarios that are supported/unsupported for Plu
75
75
76
76
### Front-end Mirroring mechanism
77
77
This section describes the front-end mirroring mechanism for python users, pointing at previous user scenarios.
78
-
***device type && subdevice type**
78
+
***Device type && Subdevice type**
79
79
Device type is user visible. User can specify the device type for the ops. e.g, "gpu", "xpu", "cpu". Subdevice type is user visible and user can specify which subdevice to use for the device type(mirroring), e.g.("NVIDIA_GPU", "INTEL_GPU", "AMD_GPU").
80
80
```
81
81
>> with tf.device("/gpu:0"):
@@ -86,32 +86,32 @@ This section describes the front-end mirroring mechanism for python users, point
86
86
***Front-end mirroring**
87
87
In the case of two GPUs in the same system, e.g. NVIDIA GPU + INTEL GPU and installing the Intel GPU plugin.
88
88
***Option 1**
89
-
Only plugged gpu device is visible, PluggableDeviceoverrides GPUDevice. If user want to use CUDA device, he need to uninstall the plugin
89
+
Only plugged gpu device is visible, PluggableDevice(INTEL GPU) overrides the default GPUDevice(NVIDIA GPU). If user want to use NVIDIA GPU, he needs to manually uninstall the plugin.
Both plugged gpu device and default gpu device are visible, but only one gpu can work at the same time, plugged gpu device is default enabled, if user want to use CUDA device, he need to call mirroring API(set_sub_device_mapping()) to switch to CUDA device.
98
+
Both plugged gpu device and default gpu device are visible, but only one gpu can work at the same time, plugged gpu device is default enabled, if user want to use NVIDIA GPU, he need to call mirroring API(set_sub_device_mapping()) to switch to NVIDIA gpu device.
physical device name is user visible. User can query the physical device name(e.g. "Titan V") for the specified device instance through [tf.config.experimental.get_device_details()](https://www.tensorflow.org/api_docs/python/tf/config/experimental/get_device_details).
0 commit comments