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.
This commit is contained in:
Szu-Kai Hsu (brucehsu) 2023-02-15 23:09:05 +00:00
parent 9ef04b2207
commit 8c8ad6e67f

13
requirements-dml.txt Normal file
View file

@ -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