在win10/win11上可以通过DirectML使用 AMD/Intel的核显或是独立显卡进行训练
首先要更新显卡驱动:
AMD驱动在这里更新,如果是核显选择具体CPU型号即可:
AMD核显驱动https://www.amd.com/en/support/download/drivers.htmlIntel CPU的驱动下载地址
Intel核显驱动https://www.intel.cn/content/www/cn/zh/search.html?ws=text#sort=relevancy&layout=table&f:@tabfilter=%5BDownloads%5D&f:@stm_10385_zh=%5B%E5%A4%84%E7%90%86%E5%99%A8%5D如果是安装TensorFlow 1.x版本可以参考微软官方方法
对应的 whl 文件在这里: