pip install pytorch-directml

详细教程：(4条消息) Windows下用amd显卡训练： Pytorch-directml 重大升级，改为pytorch插件形式，兼容更好_znsoft的博客-CSDN博客_amd显卡 pytorch

-----------------------------------------------------------------------------

以下为旧内容，依然适用，但是不推荐了。看横线以上的。

官方训练原理解释： ONNX Runtime Training Technical Deep Dive - Microsoft Tech Community

检查支持的设备

import onnxruntime as ort
ort.get_device()

ONNX运行时（ORT）能够通过优化的后端训练现有的PyTorch模型。为此，我们为pythorch引入了一个pythorch API，称为ORTTrainer，可用于将pythorch模型的训练后端（实例torch.nn.Module）切换到orttrainer。这需要对trainer代码进行一些更改，比如替换PyTorch优化器，还可以选择设置标志来启用其他特性，比如mixed-precisiontraining。下面是一个将ONNX运行时培训集成到PyTorchpre-training脚本中的示例代码片段：

注：目前的API是实验性的，预计在不久的将来会有重大变化。我们的目标是改进接口，以提供与Pythorch训练的无缝集成，这需要对用户的训练代码进行最小的更改。

import torch
...
import onnxruntime
from onnxruntime.capi.ort_trainer import IODescription, ModelDescription, ORTTrainer# Model definition
class Net(torch.nn.Module):def __init__(self, D_in, H, D_out):...def forward(self, x):...model = Net(D_in, H, H_out)
criterion = torch.nn.Functional.cross_entropy
description = ModelDescription(...)
optimizer = 'SGDOptimizer'
trainer = ORTTrainer(model, criterion, description, optimizer, ...)# Training Loop
for t in range(1000):# forward + backward + weight updateloss, y_pred = trainer.train_step(x, y, learning_rate)...

Windows 平台下AMD 显卡加速pytorch训练相关推荐

windows平台下vlc编译之十三：vlc1.1.0编译
请移步https://higoge.github.io/,所有下载资料在那个博客都能找到.谢谢. --------------------------------------------------- ...
windows平台下vlc编译之六：vlc-0.9.8a的编译
请移步https://higoge.github.io/,所有下载资料在那个博客都能找到.谢谢. --------------------------------------------------- ...
Windows平台下Makefile学习笔记
来源:http://blog.csdn.net/clever101 决心学习Makefile,一方面是为了解决编译开源代码时需要跨编译平台的问题(发现一些开源代码已经在使用VS2010开发,但我还没安 ...
windows平台下vlc编译
转自:http://jeremiah.blog.51cto.com/539865/114190 Jeremiah刚刚工作几个月,参与的第一个项目是与视频监控有关,分配给我的任务就是用开源的vl ...
Windows平台下Git服务器搭建
初识Git Git是什么? Git是目前世界上最先进的分布式版本控制系统. SVN与Git的最主要的区别? SVN是集中式版本控制系统,版本库是集中放在中央服务器的,而干活的时候,用的都是自己的电脑, ...
Windows平台下的session0创建进程的问题与解决办法
Windows平台下的session0创建进程的问题与解决办法参考文章: (1)Windows平台下的session0创建进程的问题与解决办法 (2)https://www.cnblogs.com/ ...
Windows平台下NS2网络仿真环境的搭建
NS2(Network Simulator 2) 是一种针对网络技术的源代码公开的.免费的软件模拟平台,研究人员使用它可以很容易的进行网络技术的开发,而且发展到今天,它所包含的模块几乎涉及到了网络技术 ...
windows平台下的oracle ORA-01031的解决方法
windows平台下的oracle ORA-01031的解决方法今天下午遇到一个很怪异的问题,在windows平台下sqlplus / as sysdba登陆数据库,提示权限不足, 当时就纳闷了,s ...
Windows平台下Glade+GTK开发环境的搭建
[@.1 MVVM设计模式与Glade] 做上层软件开发的程序员可能对于MVVM模式比较熟悉,这是一种经典的软件设计模式,很好的将用户界面与后台处理之间分层开,通过属性.事件绑定这种统一的" ...

Windows 平台下AMD 显卡加速pytorch训练

pip install pytorch-directml

Windows 平台下AMD 显卡加速pytorch训练相关推荐

最新文章

热门文章