TensorFlow/PyTorch 使用核显训练(仅限win平台)

本文介绍了如何在Windows10/11系统上通过DirectML利用AMD/Intel显卡进行TensorFlow和PyTorch的训练,包括驱动更新、不同显卡的性能比较以及多GPU的设置方法。结果显示,DirectML接口相比cuDNN约慢30%。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

在win10/win11上可以通过DirectML使用 AMD/Intel的核显或是独立显卡进行训练

首先要更新显卡驱动:

AMD驱动在这里更新,如果是核显选择具体CPU型号即可:

AMD核显驱动icon-default.png?t=N7T8https://www.amd.com/en/support/download/drivers.htmlIntel CPU的驱动下载地址

Intel核显驱动icon-default.png?t=N7T8https://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版本可以参考微软官方方法

在 Windows 上通过 DirectML 启用 TensorFlow1.xicon-default.png?t=N7T8https://learn.microsoft.com/zh-cn/windows/ai/directml/gpu-tensorflow-windows

对应的 whl 文件在这里:

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值