Windows下用amd显卡训练： Pytorch-directml 重大升级，改为pytorch插件形式，兼容更好

znsoft

已于 2022-12-22 07:49:52 修改

阅读量7.2k

点赞数 3

分类专栏： pytorch 文章标签： pytorch 深度学习 python

于 2022-12-22 07:49:15 首次发布

本文链接：https://blog.csdn.net/znsoft/article/details/128403947

版权

pytorch 专栏收录该内容

11 篇文章

订阅专栏

安装方法：

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()