Pytorch 打包 EXE: 将 Pytorch 项目转换为独立的可执行文件
在开发过程中，我们可能想将我们的 Pytorch 项目打包为一个独立的可执行文件，以便其他人可以在没有 Python 和 Pytorch 环境的情况下轻松运行它。这个过程有很多方法，但接下来我们将介绍如何使用 Pyinstaller 将 Pytorch 项目打包为一个可执行文件 (EXE)。
首先，我们需要确保已经创建了一个适用于该项目的虚拟环境。这将使得项目打包更容易，因为它将只包含需要的包。在项目所在的文件夹中创建一个虚拟环境：
```bash
python -m venv venv
```
激活虚拟环境并安装 Pytorch 和其他依赖项：
```bash
# For Windows:
venv\Scripts\activate
# For Linux and macOS:
source venv/bin/activate
pip install torch torchvision
```
安装 Pyinstaller:
```bash
pip install pyinstaller
```
现在，我们可以使用 Pyinstaller 打包项目：
```bash
pyinstaller --onefile your_pytorch_script.py
```
这将创建一个名为 `dist` 的新目录，其中包含单个可执行文件。但是，由于 PyTorch 使用许多动态库，我们需要添加一些运行时HOOK。
创建一个名为 `hook-torch.py` 的新文件，添加以下代码：
```python
from PyInstaller.utils.hooks import collect_all
torch_root = 'your_custom_torch_path' # You need to specify path to your Pytorch installation
# ----------------------------- CUT HERE -----------------------------
datas, binaries, hiddenimports = collect_all('torch')
```
在项目中创建一个名为 hooks 的文件夹，并将 hook-torch.py 文件移到其中。
找到你的环境中的 Pytorch 安装位置。通常，您可以在 `venv\Lib\site-packages` (Windows) 或 `venv/lib/pythonX.Y/site-packages` (Linux/macOS) 中找到它。
现在，运行以下命令打包项目，相应地替换 `your_pytorch_script.py` 和 `your_custom_torch_path`：
```bash
pyinstaller --onefile --additional-hooks-dir=hooks --hidden-import=torch._C --exclude-module=torch._C --exclude-module=torch._six --exclude-module=torch.multiprocessing.spawn --exclude-module=torch.package --exclude-module=torch.quantization.cxx -r "hook-torch.py:your_custom_torch_path/lib/*.dll" your_pytorch_script.py
```
稍等片刻，可执行文件将在 `dist` 文件夹中生成。
请注意，生成的 EXE 文件可能会变得相当大，但这通常是因为它包含了 PyTorch 和其他所有依赖项。此外，由于可执行文件的特点，一些防病毒软件可能误报为病毒，但这是预期行为。可以添加数字签名或将可执行文件加入白名单以解决此问题。
现在您可以将生成的 EXE 文件发送给其他人，他们可以在没有安装 Python 和 Pytorch 的情况下轻松运行您的项目。
