确保你安装了正确的 torch 和 torchvision 版本。建议使用pip安装torch和torchvision。
确保你的代码中引用了正确的 torch 和 torchvision 包,且版本与你安装的版本相同。
选择合适的Pyinstaller指令。建议在使用 auto-py-to-exe 进行打包前,对 Pyinstaller 进行配置优化。这样可以更好的解决这个问题,同时获得更好的打包结果。
以下是使用 Pyinstaller 手动打包 torch 和 torchvision 的示例代码:
pyinstaller --hidden-import torch --hidden-import torchvision your_script.py
其中, your_script.py
是你要打包的脚本, --hidden-import
选项告诉 Pyinstaller 导入必要的 torch 和 torchvision 包。
如果你使用的是 anaconda 虚拟环境,可以使用以下示例代码:
pyinstaller --hidden-import torch --hidden-import torchvision --additional-hooks-dir /path/to/anaconda3/envs/your_env_name/hooks your_script.py
其中, --additional-hooks-dir
选项告诉 Pyinstaller 查找必要的环境钩子。
(注意:示例代码中的 /path/to/anaconda3/envs/your_env_name/hooks
应更改为你 anaconda 环境的实际路径。)