When i test the torch.hub.load some error

username: lonny
cooked: <p>my code is</p>
<pre><code class="lang-auto">import torch
repo = 'pytorch/vision'
model = torch.hub.load(repo, 'resnet50', pretrained=True)
</code></pre>
<p>the error is<br>
<code>ImportError: cannot import name 'get_weight' from 'torchvision.models' (C:\ProgramData\Anaconda3\lib\site-packages\torchvision\models\__init__.py)</code></p>
<p>what’s wrong<br>
is the program updated?</p>

username: ptrblck
cooked: <p>I cannot reproduce the error and get a valid model using the latest PyTorch and <code>torchvision</code> nightly binaries. Are you mixing different PyTorch and <code>torchvision</code> installations?</p>

username: lonny
cooked: <p>i think so , this error looks like the package version is wrong, but i install the use conda, maybe i  should check it.</p>

username: lonny
cooked: <p>i have checked my version on conda, the PyTorch is 1.11.0 and the torchvision is 0.12.0<br>
should i change it?</p>

username: ptrblck
cooked: <p>I can reproduce it in these versions and it seems that the latest <a href="https://github.com/pytorch/vision/blob/5985504cc32011fbd4312600b4492d8ae0dd13b4/hubconf.py#L4"><code>hubconf.py</code></a> file is being downloaded, which has recently (3 days ago) added the <code>get_weight</code> import.<br>
I don’t quite understand the workflow as I would have assumed the packaged <code>hubconf.py</code> would be used, but I see that the one from the <code>vision/main</code> is being downloaded to the <code>.cache</code> directory.<br>
Even after removing the <code>import</code> the script will fail with imports of unknown models which were not available in these stable releases.</p>
<p>CC <a class="mention" href="/u/pmeier">@pmeier</a> is this behavior expected?</p>

username: lonny
cooked: <p>that’s true , i don’t know how to fix it</p>

username: ptrblck
cooked: <p>As a workaround you could install the latest nightly releases which I would assume should work.</p>

username: Image_Scientist
cooked: <p>Hi! I tried this but have not had any luck.</p>
<pre><code class="lang-auto">model = torch.hub.load('pytorch/vision', 'resnet18', pretrained=True)
</code></pre>
<p>Using <a href="https://anaconda.org/pytorch-nightly/torchvision" class="inline-onebox" rel="noopener nofollow ugc">Torchvision :: Anaconda.org</a> to update to the nightly build does not change my versions of PyTorch or Torchvision</p>
<p>Both are what are listed above - torchvision 0.12.0 and PyTorch 1.11.0.</p>
<details>
<summary>
my environment</summary>
<pre><code class="lang-auto"># packages in environment at D:\Anaconda\envs\fastai:
#
# Name                    Version                   Build  Channel
anyio                     3.6.1                    pypi_0    pypi
argon2-cffi               21.3.0             pyhd3eb1b0_0
argon2-cffi-bindings      21.2.0          py310h2bbff1b_0
asttokens                 2.0.5              pyhd3eb1b0_0
attrs                     21.4.0             pyhd3eb1b0_0
babel                     2.10.3                   pypi_0    pypi
backcall                  0.2.0              pyhd3eb1b0_0
beautifulsoup4            4.11.1          py310haa95532_0
blas                      1.0                         mkl
bleach                    4.1.0              pyhd3eb1b0_0
blosc                     1.21.0               h19a0ad4_0
bottleneck                1.3.4           py310h9128911_0
brotli                    1.0.9                ha925a31_2
brotlipy                  0.7.0           py310h2bbff1b_1002
bzip2                     1.0.8                he774522_0
ca-certificates           2022.4.26            haa95532_0
catalogue                 2.0.7           py310haa95532_0
certifi                   2022.6.15       py310haa95532_0
cffi                      1.15.0          py310h2bbff1b_1
cfitsio                   3.470                he774522_6
charls                    2.2.0                h6c2663c_0
charset-normalizer        2.0.4              pyhd3eb1b0_0
click                     8.0.4           py310haa95532_0
cloudpickle               2.0.0              pyhd3eb1b0_0
colorama                  0.4.4              pyhd3eb1b0_0
cryptography              37.0.1          py310h21b164f_0
cudatoolkit               11.3.1               h59b6b97_2
cycler                    0.11.0             pyhd3eb1b0_0
cymem                     2.0.6           py310hd77b12b_0
cython-blis               0.7.7           py310h9128911_0
cytoolz                   0.11.0          py310h2bbff1b_0
dask-core                 2022.5.0        py310haa95532_0
debugpy                   1.5.1           py310hd77b12b_0
decorator                 5.1.1              pyhd3eb1b0_0
defusedxml                0.7.1              pyhd3eb1b0_0
entrypoints               0.4             py310haa95532_0
executing                 0.8.3              pyhd3eb1b0_0
fastai                    2.6.3                      py_0    fastai
fastbook                  0.0.24                     py_0    fastai
fastcore                  1.4.4                      py_0    fastai
fastdownload              0.0.6                      py_0    fastai
fastprogress              1.0.2                      py_0    fastai
fonttools                 4.25.0             pyhd3eb1b0_0
freetype                  2.10.4               hd328e21_0
fsspec                    2022.3.0        py310haa95532_0
giflib                    5.2.1                h62dcd97_0
graphviz                  2.38                 hfd603c8_2
icc_rt                    2019.0.0             h0cc432a_1
icu                       58.2                 ha925a31_3
idna                      3.3                pyhd3eb1b0_0
imagecodecs               2021.8.26       py310h6cc35ac_0
imageio                   2.9.0              pyhd3eb1b0_0
intel-openmp              2021.4.0          haa95532_3556
ipykernel                 6.9.1           py310haa95532_0
ipython                   8.3.0           py310haa95532_0
ipython_genutils          0.2.0              pyhd3eb1b0_1
ipywidgets                7.6.5              pyhd3eb1b0_1
jedi                      0.18.1          py310haa95532_1
jinja2                    3.0.3              pyhd3eb1b0_0
joblib                    1.1.0              pyhd3eb1b0_0
jpeg                      9e                   h2bbff1b_0
json5                     0.9.8                    pypi_0    pypi
jsonschema                4.4.0           py310haa95532_0
jupyter-server            1.17.1                   pypi_0    pypi
jupyter_client            7.2.2           py310haa95532_0
jupyter_core              4.10.0          py310haa95532_0
jupyterlab-server         2.14.0                   pypi_0    pypi
jupyterlab_pygments       0.1.2                      py_0
jupyterlab_widgets        1.0.0              pyhd3eb1b0_1
kiwisolver                1.4.2           py310hd77b12b_0
langcodes                 3.3.0              pyhd3eb1b0_0
lcms2                     2.12                 h83e58a3_0
lerc                      3.0                  hd77b12b_0
libaec                    1.0.4                h33f27b4_1
libdeflate                1.8                  h2bbff1b_5
libffi                    3.4.2                h604cdb4_1
libpng                    1.6.37               h2a8f88b_0
libtiff                   4.2.0                hd0e1b90_0
libuv                     1.40.0               he774522_0
libwebp                   1.2.2                h2bbff1b_0
libzopfli                 1.0.3                ha925a31_0
locket                    1.0.0           py310haa95532_0
lz4-c                     1.9.3                h2bbff1b_1
markupsafe                2.0.1           py310h2bbff1b_0
matplotlib                3.5.1           py310haa95532_1
matplotlib-base           3.5.1           py310hd77b12b_1
matplotlib-inline         0.1.2              pyhd3eb1b0_2
mistune                   0.8.4           py310h2bbff1b_1000
mkl                       2021.4.0           haa95532_640
mkl-service               2.4.0           py310h2bbff1b_0
mkl_fft                   1.3.1           py310ha0764ea_0
mkl_random                1.2.2           py310h4ed8f06_0
munkres                   1.1.4                      py_0
murmurhash                1.0.7           py310hd77b12b_0
nbclient                  0.5.13          py310haa95532_0
nbconvert                 6.5.0                    pypi_0    pypi
nbformat                  5.3.0           py310haa95532_0
nest-asyncio              1.5.5           py310haa95532_0
networkx                  2.7.1              pyhd3eb1b0_0
notebook                  6.4.11          py310haa95532_0
numexpr                   2.8.1           py310hb57aa6b_1
numpy                     1.22.3          py310h6d2d95c_0
numpy-base                1.22.3          py310h206c741_0
openjpeg                  2.4.0                h4fc8c34_0
openssl                   1.1.1o               h2bbff1b_0
packaging                 21.3               pyhd3eb1b0_0
pandas                    1.4.2           py310hd77b12b_0
pandocfilters             1.5.0              pyhd3eb1b0_0
parso                     0.8.3              pyhd3eb1b0_0
partd                     1.2.0              pyhd3eb1b0_1
pathy                     0.6.1           py310haa95532_0
pickleshare               0.7.5           pyhd3eb1b0_1003
pillow                    9.0.1           py310hdc2b20a_0
pip                       21.2.4          py310haa95532_0
preshed                   3.0.6           py310h6c2663c_0
prometheus_client         0.13.1             pyhd3eb1b0_0
prompt-toolkit            3.0.20             pyhd3eb1b0_0
pure_eval                 0.2.2              pyhd3eb1b0_0
pycparser                 2.21               pyhd3eb1b0_0
pydantic                  1.9.0           py310h2bbff1b_1
pygments                  2.11.2             pyhd3eb1b0_0
pyopenssl                 22.0.0             pyhd3eb1b0_0
pyparsing                 3.0.4              pyhd3eb1b0_0
pyqt                      5.9.2           py310hd77b12b_6
pyrsistent                0.18.0          py310h2bbff1b_0
pysocks                   1.7.1           py310haa95532_0
python                    3.10.4               hbb2ffb3_0
python-dateutil           2.8.2              pyhd3eb1b0_0
python-fastjsonschema     2.15.1             pyhd3eb1b0_0
python-graphviz           0.16               pyhd3eb1b0_1
pytorch                   1.11.0          py3.10_cuda11.3_cudnn8_0    pytorch
pytorch-mutex             1.0                        cuda    pytorch
pytz                      2021.3             pyhd3eb1b0_0
pywavelets                1.3.0           py310h2bbff1b_0
pywin32                   302             py310h2bbff1b_2
pywinpty                  2.0.2           py310h5da7b33_0
pyyaml                    6.0             py310h2bbff1b_0
pyzmq                     22.3.0          py310hd77b12b_2
qt                        5.9.7            vc14h73c81de_0
requests                  2.27.1             pyhd3eb1b0_0
scikit-image              0.19.2          py310h4ed8f06_0
scikit-learn              1.0.2           py310h4ed8f06_0
scipy                     1.7.3           py310h6d2d95c_0
send2trash                1.8.0              pyhd3eb1b0_1
sentencepiece             0.1.96                   pypi_0    pypi
setuptools                61.2.0          py310haa95532_0
shellingham               1.3.1              pyhd3eb1b0_0
sip                       4.19.13         py310hd77b12b_0
six                       1.16.0             pyhd3eb1b0_1
smart_open                5.2.1           py310haa95532_0
snappy                    1.1.9                h6c2663c_0
sniffio                   1.2.0                    pypi_0    pypi
soupsieve                 2.3.1              pyhd3eb1b0_0
spacy                     3.3.0           py310hef0f399_0
spacy-legacy              3.0.9           py310haa95532_0
spacy-loggers             1.0.1              pyhd3eb1b0_0
sqlite                    3.38.3               h2bbff1b_0
srsly                     2.4.3           py310hd77b12b_0
stack_data                0.2.0              pyhd3eb1b0_0
terminado                 0.13.1          py310haa95532_0
testpath                  0.5.0              pyhd3eb1b0_0
thinc                     8.0.15          py310hf497b98_0
threadpoolctl             2.2.0              pyh0d69192_0
tifffile                  2021.7.2           pyhd3eb1b0_2
tinycss2                  1.1.1                    pypi_0    pypi
tk                        8.6.12               h2bbff1b_0
toolz                     0.11.2             pyhd3eb1b0_0
torchaudio                0.11.0              py310_cu113    pytorch
torchvision               0.12.0              py310_cu113    pytorch
tornado                   6.1             py310h2bbff1b_0
tqdm                      4.64.0          py310haa95532_0
traitlets                 5.1.1              pyhd3eb1b0_0
typer                     0.4.1           py310haa95532_0
typing-extensions         4.1.1                hd3eb1b0_0
typing_extensions         4.1.1              pyh06a4308_0
tzdata                    2022a                hda174b7_0
urllib3                   1.26.9          py310haa95532_0
vc                        14.2                 h21ff451_1
voila                     0.3.5                    pypi_0    pypi
vs2015_runtime            14.27.29016          h5e58377_2
wasabi                    0.9.1           py310haa95532_0
wcwidth                   0.2.5              pyhd3eb1b0_0
webencodings              0.5.1           py310haa95532_1
websocket-client          1.3.3                    pypi_0    pypi
websockets                10.3                     pypi_0    pypi
wheel                     0.37.1             pyhd3eb1b0_0
widgetsnbextension        3.5.2           py310haa95532_0
win_inet_pton             1.1.0           py310haa95532_0
wincertstore              0.2             py310haa95532_2
winpty                    0.4.3                         4
xz                        5.2.5                h8cc25b3_1
yaml                      0.2.5                he774522_0
zfp                       0.5.5                hd77b12b_6
zlib                      1.2.12               h8cc25b3_2
zstd                      1.4.9                h19a0ad4_0
</code></pre>
</details>
<pre><code class="lang-auto">----&gt; 4 from torchvision.models import get_weight
      5 from torchvision.models.alexnet import alexnet
      6 from torchvision.models.convnext import convnext_tiny, convnext_small, convnext_base, convnext_large

ImportError: cannot import name 'get_weight' from 'torchvision.models' (d:\Anaconda\envs\fastai2\lib\site-packages\torchvision\models\__init__.py)
</code></pre>
<p>PyTorch nightly did not change anything, All requested packages already installed. Torchvision nightly only added <code>nbconvert</code> and updated <code>certifi</code></p>
<p>Edit: Nevermind - sort of. I found out that it does work in Jupyter Notebook in Chrome in that environment - however in VS Code using the same environment it fails with the given error message.</p>

username: ptrblck
cooked: <p>Most likely your VS Code setup is using another environment and is thus failing while the Jupyter notebook uses the newer (and fixed) one.</p>

username: Image_Scientist
cooked: <p>I made sure they were both using the same environment, and it was a fresh environment so there was no chance that each was using a slightly different version. For the moment I can use Jupyter Notebooks I guess. And it works fine on other models, I just can’t load specific models like ResNet18.</p>
<pre><code class="lang-auto">model = torch.hub.load('mateuszbuda/brain-segmentation-pytorch', 'unet', in_channels=1, out_channels=1, init_features=8, pretrained=False)
</code></pre>
<p>for example works in either.</p>

username: oke-aditya
cooked: <p>Hi. Torchvision models have changed since torchvision 0.13 released few days ago.<br>
Can you try with torchvision 0.13?</p>

username: Image_Scientist
cooked: <aside class="quote no-group" data-username="Image_Scientist" data-post="8" data-topic="152054">
<div class="title">
<div class="quote-controls"></div>
<img loading="lazy" alt="" width="24" height="24" src="https://discuss.pytorch.org/user_avatar/discuss.pytorch.org/image_scientist/48/50860_2.png" class="avatar"> Image_Scientist:</div>
<blockquote>
<p><code>model = torch.hub.load('pytorch/vision', 'resnet18', pretrained=True)</code></p>
</blockquote>
</aside>
<p>Sorry, just got back to this. I can confirm that updating to torchvision 0.13.0 resolves the issue.</p>
<pre><code class="lang-auto">pip install torchvision==0.13.0 --user
</code></pre>
<p>–user was necessary for this particular update on my system for some reason, though it is not usually.</p>