《Nuitka打包实战指南》实战打包PyTorch

Posted 2023-01-06 la_vie_est_belle

实战打包PyTorch

打包示例源码：

请看文章末尾

版本信息：

torch==1.10.2

Nuitka==0.6.19.1

打包系统：

 Windows10 64位

打包前我们需要运行下代码，确保没有报错。

hello.py代码如下：

import torch
from torch import nn


# Get cpu or gpu device for training.
device = "cuda" if torch.cuda.is_available() else "cpu"
print(f"Using device device")


# Define model
class NeuralNetwork(nn.Module):
    def __init__(self):
        super(NeuralNetwork, self).__init__()
        self.flatten = nn.Flatten()
        self.linear_relu_stack = nn.Sequential(
            nn.Linear(28*28, 512),
            nn.ReLU(),
            nn.Linear(512, 512),
            nn.ReLU(),
            nn.Linear(512, 10)
        )

    def forward(self, x):
        x = self.flatten(x)