admin管理员组

文章数量:1530021

安装方法：

Enable PyTorch with DirectML on Windows | Microsoft Learn

新的pytorch-directml 不再是独立的pytorch 移植，变成了一个设备插件，更好用，兼容性更好。

原来的版本无法跑transformers, 新版变成一个独立的计算设备 dml, 兼容性更好

import torch
import torch_directml


dml=torch_directml.device()
tensor1 = torch.tensor([1]).to(dml) # Note that dml is a variable, not a string!
tensor2 = torch.tensor([2]).to(dml)

for i in range(1000000):
    dml_algebra = tensor1 + tensor2
    dml_algebra.item()

本文标签： 插件显卡形式Windowsamd

版权声明：本文标题：Windows下用amd显卡训练 ： Pytorch-directml 重大升级，改为pytorch插件形式，兼容更好 内容由热心网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：https://m.elefans.com/dianzi/1726770899a1083860.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。