11:27:18.337830: I tensorflow/core/common_runtime/eager/:571] Executing op Add in device /job:localhost/replica:0/task:0/device:DML:0 11:27:18.240065: I tensorflow/core/common_runtime/dml/dml_device_:32] DirectML: creating device on adapter 0 (AMD Radeon VII) 11:27:18.235973: I tensorflow/core/common_runtime/dml/dml_device_:45] DirectML device enumeration: found 1 compatible adapters. You should see output similar to the following, with the add operator placed on the DML device. Tf.enable_eager_execution(tf.ConfigProto(log_device_placement=True)) Copy the following lines into an interactive Python session. Once you've installed the tensorflow-directml package, you can verify that it runs correctly by adding two tensors. The tensorflow-directml package supports only TensorFlow 1.15. Once Miniconda is installed, create an environment using Python named directml, and activate it through the following commands.

There is additional guidance for setup on Anaconda's site. Set up Python environmentĭownload and install the Miniconda Windows installer on your system.

The rest of this setup assumes that you use a Miniconda environment. There are many tools that you can use to set up a virtual Python environment-for these instructions, we'll use Anaconda's Miniconda. We recommend setting up a virtual Python environment inside Windows. Select Check for updates in the Windows Update section of the Settings app. Check for GPU driver updatesĮnsure that you have the latest GPU driver installed. You can check your build version number by running winver via the Run command (Windows logo key + R). The TensorFlow with DirectML package on native Windows works starting with Windows 10, version 1709 (Build 16299 or higher).

Once set up, you can use your exisiting model scripts or check out a few samples on the DirectML repo. To use DirectML on TensorFlow 2, check out the TensorFlow-DirectML-Plugin. This release provides students, beginners, and professionals a way to run machine learning (ML) training on their existing hardware by using the TensorFlow with DirectML package for TensorFlow 1.15.

Install conda windows