Model takes very long to load and tutorial script fails

[DragonDuck]

Model takes very long to load and tutorial script fails

[Zehaos]

The version of cudatoolkit is 10.1.168, while pytorch1.3 is build with cuda 10.1.243. Maybe this is the problem live in.

[ppwwyyxx]

I think you can easily verify that it does not take more than 1 minute on colab, so this should be an environment issue.

What looks problematic in your environment is that your version of pytorch is not built with the compute compatibility for P100. In that case when some ops are run for the first time nvidia-driver will spend some time compiling them. Running it the second time on the same machine should be faster.

Running the code you showed also needs to download 178MB of model for the first time. This is fast on colab, but may be slow on your machine.

AttributeError: 'NoneType' object has no attribute 'shape'

It is saying that opencv cannot read the input image you gave. You may need to check your input path or the opencv installation.

[ppwwyyxx]

Similar reports have been seen in #27 and pytorch/pytorch#537. You need to find a version of pytorch whose "NVCC architecture flags" include the compute compatibility of your GPU.
Closing as the issue is about pytorch installation.

[ppwwyyxx]

Has been fixed in pytorch according to #27

[DragonDuck]

Thank you very much for the help! I can confirm that the Conda version of PyTorch from last week wasn't properly compiled to support my GPU. This has been fixed and the newest PyTorch version downloaded via Conda works error-free.