4/4/2023 0 Comments Metal supported gpuOn the new APU device: import tensorflow as tf # TensorFlow registers PluggableDevices here. With the plug-in installed, test that the device is visible and run an operation Successfully installed tensorflow-apu-0.0.1 Plug-in only has one custom kernel for ReLU: # Install the APU example plug-in package Processing Unit (APU), is installed and used. The followingĬode snippet shows how the plugin for a new demonstration device, Awesome Only have to install the device plug-in package for that device. To use a particular device, like one would a native device in TensorFlow, users Plug-in developers maintain separate code repositories and distribution packagesįor their plugins and are responsible for testing their devices. Relies on C APIs to communicate with the TensorFlow binary in a stable manner. The mechanism requires no device-specific changes in the TensorFlow code. Installed alongside the official TensorFlow package. For NVIDIA® GPU support, go toĪrchitecture adds new device support as separate plug-in packages that are Note: This page is for non-NVIDIA® GPU devices.
0 Comments
Leave a Reply. |
AuthorWrite something about yourself. No need to be fancy, just an overview. ArchivesCategories |