From 8c8ad6e67fe36207df12c6dcf62c1354ff5fe5ee Mon Sep 17 00:00:00 2001 From: "Szu-Kai Hsu (brucehsu)" Date: Wed, 15 Feb 2023 23:09:05 +0000 Subject: [PATCH] feat: add DirectML plugin support in Tensorflow 2. ... tensorflow 2.10.0 is used since it'll be the last version with native Windows support. WSL2 is required from 2.11.0. --- requirements-dml.txt | 13 +++++++++++++ 1 file changed, 13 insertions(+) create mode 100644 requirements-dml.txt diff --git a/requirements-dml.txt b/requirements-dml.txt new file mode 100644 index 0000000..5810766 --- /dev/null +++ b/requirements-dml.txt @@ -0,0 +1,13 @@ +tqdm +numpy==1.20.1 +numexpr +h5py==2.10.0 +opencv-python==4.3.0.38 +ffmpeg-python==0.1.17 +scikit-image==0.14.2 +scipy==1.4.1 +colorama +tensorflow-cpu==2.10.0 +pyqt5 +tf2onnx +tensorflow-directml-plugin \ No newline at end of file