6

这是我的main.py

import torchvision
input("Press key")

它在命令行中正确运行:python main.py

我需要一个适用于 Windows 的可执行文件。所以我做了 :pyinstaller main.py

但是当我启动main.exe时,里面/dist/main出现了这个错误:

Traceback (most recent call last):
  File "main.py", line 1, in <module>

  ... (omitted)

  File "site-packages\torchvision\ops\misc.py", line 135, in <module>
  File "site-packages\torchvision\ops\misc.py", line 148, in FrozenBatchNorm2d
  File "site-packages\torch\jit\__init__.py", line 850, in script_method
  File "site-packages\torch\jit\frontend.py", line 152, in get_jit_def
  File "inspect.py", line 973, in getsource
  File "inspect.py", line 955, in getsourcelines
  File "inspect.py", line 786, in findsource
OSError: could not get source code
[2836] Failed to execute script main

似乎某些源代码没有从 pyinstaller 正确导入。我不确定问题是torch模块还是torchvision

附加信息:

  • 我最近安装了 Visual Studio 2019

系统信息:

  • 窗口 10
  • 蟒蛇 3.7
  • 火炬-1.1.0
  • 火炬视觉-0.3.0

[编辑]

我发现问题出在torchvision中的FrozenBatchNorm2d 类的定义中。以下脚本产生与之前发布的相同的错误:

主文件

import torch

class FrozenBatchNorm2d(torch.jit.ScriptModule):

    def __init__(self, n):
        super(FrozenBatchNorm2d, self).__init__()

    @torch.jit.script_method

    def forward(self):
        pass

我复制了所有的手电筒源文件。但我仍然得到错误...

4

3 回答 3

3

将torchvision降级到以前的版本修复错误。

pip uninstall torchvision
pip install torchvision==0.2.2.post3
于 2019-05-28T08:58:09.880 回答
1

试试这个猴子补丁。

def script_method(fn, _rcb=None):
    return fn
def script(obj, optimize=True, _frames_up=0, _rcb=None):
    return obj    
import torch.jit
torch.jit.script_method = script_method 
torch.jit.script = script
于 2020-01-20T11:02:57.880 回答
1

猴子补丁对我不起作用,因为我在导入torch.jit 时出错。

因此,在导入 torch 之前,我在 main.py 中使用了以下解决方法:

os.environ["PYTORCH_JIT"] = "0"

显然,您失去了 JIT 优化,但至少可执行文件有效。

于 2020-05-12T06:56:00.473 回答