核显 DirectML 深度学习环境 2022-10-05 参考文档: tensorflow-directml 官方文档 Enable TensorFlow with DirectML in WSL 环境: win10 + wsl2-ubuntu20.04 安装显卡驱动(确保型号匹配,我是 amd 的 r5 核显): amd:见 Radeon™ Software Adrenalin 2020 Edition for Microsoft® DirectX on Windows Subsystem for Linux Release Notes 支持的型号,只要大于 20.20.01.05 driver or newer 即可。 intel: 6xx 系 HD 核显 以及更高的版本。28.20.100.8322 driver or newer Nvida? cuda 不香吗。 py36 环境安装 截至 2022 年 10 月 5 日, tensorflow-directml==1.15.8 最高支持 py37,但是其依赖 gast 0.2.2 并不支持 pep517 的方式安装,推荐用 py36 版本。 若已有 py3.6 ....