Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

TypeError: 'int' object is not callable during torch model weights load #8

Open
MrCsabaToth opened this issue Feb 17, 2025 · 4 comments
Open

TypeError: 'int' object is not callable during torch model weights load #8

MrCsabaToth opened this issue Feb 17, 2025 · 4 comments

Comments

@MrCsabaToth
Copy link

I followed the README.

  1. Setup the conda env based on the environment.yaml
  2. Downloaded the weights.tar and also the two npy stat files
  3. Compiled and installed wgrib2
  4. Used gdas script to download data for a 2023 summer day
  5. Trying to call inference. I checked that the file is in the right place. With unzip -l it seems it's not corrupted but who knows.
fourcastnet) jupyter@instance-20250216-164054:~/FourCastNet/NCEP$ python inference.py 2023080400 -w model/ -i input_2023080400.npy -o output/ -l 336
model/weights.tar
Traceback (most recent call last):
  File "/home/jupyter/FourCastNet/NCEP/inference.py", line 265, in <module>
    fcn.run()
    ~~~~~~~^^
  File "/home/jupyter/FourCastNet/NCEP/inference.py", line 196, in run
    model = self.load_model(path)
  File "/home/jupyter/FourCastNet/NCEP/inference.py", line 158, in load_model
    checkpoint = torch.load(checkpoint_file, map_location=device)
  File "/opt/conda/envs/fourcastnet/lib/python3.13/site-packages/torch/serialization.py", line 1462, in load
    return _load(
        opened_zipfile,
    ...<3 lines>...
        **pickle_load_args,
    )
  File "/opt/conda/envs/fourcastnet/lib/python3.13/site-packages/torch/serialization.py", line 1964, in _load
    result = unpickler.load()
  File "/opt/conda/envs/fourcastnet/lib/python3.13/site-packages/torch/_weights_only_unpickler.py", line 512, in load
    self.append(self.persistent_load(pid))
                ~~~~~~~~~~~~~~~~~~~~^^^^^
  File "/opt/conda/envs/fourcastnet/lib/python3.13/site-packages/torch/serialization.py", line 1928, in persistent_load
    typed_storage = load_tensor(
        dtype, nbytes, key, _maybe_decode_ascii(location)
    )
  File "/opt/conda/envs/fourcastnet/lib/python3.13/site-packages/torch/serialization.py", line 1900, in load_tensor
    wrap_storage=restore_location(storage, location),
                 ~~~~~~~~~~~~~~~~^^^^^^^^^^^^^^^^^^^
  File "/opt/conda/envs/fourcastnet/lib/python3.13/site-packages/torch/serialization.py", line 1811, in restore_location
    result = map_location(storage, location)
TypeError: 'int' object is not callable

This is a blocker right now.
@MrCsabaToth
Copy link
Author

Unfortunately it seems that in the guts there's unpickling going on. That means potential version mismatch between some libraries at the time of pickle vs unpickle can cause misalignment?

@MrCsabaToth
Copy link
Author

MrCsabaToth commented Feb 17, 2025 •
edited
Loading

What was the pytorch version during pickle? Mine is 2.6.0

(fourcastnet) jupyter@instance-20250216-164054:~/FourCastNet/NCEP/model$ conda list
# packages in environment at /opt/conda/envs/fourcastnet:
#
# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                 conda_forge    conda-forge
_openmp_mutex             4.5                  2_kmp_llvm    conda-forge
ai-models                 0.7.4                    pypi_0    pypi
ai-models-fourcastnetv2   0.0.3                    pypi_0    pypi
array-api-compat          1.10.0                   pypi_0    pypi
attr                      2.5.1                h166bdaf_1    conda-forge
attrs                     25.1.0             pyh71513ae_0    conda-forge
blosc                     1.21.6               he440d0b_1    conda-forge
boto3                     1.36.21            pyhd8ed1ab_0    conda-forge
botocore                  1.36.21         pyge310_1234567_0    conda-forge
brotli                    1.1.0                hb9d3cd8_2    conda-forge
brotli-bin                1.1.0                hb9d3cd8_2    conda-forge
brotli-python             1.1.0           py313h46c70d0_2    conda-forge
bzip2                     1.0.8                h4bc722e_7    conda-forge
c-ares                    1.34.4               hb9d3cd8_0    conda-forge
ca-certificates           2025.1.31            hbcca054_0    conda-forge
cartopy                   0.24.0          py313ha87cce1_0    conda-forge
cdsapi                    0.7.5                    pypi_0    pypi
certifi                   2025.1.31          pyhd8ed1ab_0    conda-forge
cf-units                  3.3.0           py313ha014f3b_0    conda-forge
cffi                      1.17.1          py313hfab6e84_0    conda-forge
cfgrib                    0.9.15.0                 pypi_0    pypi
cftime                    1.6.4           py313ha014f3b_1    conda-forge
cfunits                   3.3.7              pyhd8ed1ab_1    conda-forge
charset-normalizer        3.4.1                    pypi_0    pypi
click                     8.1.8              pyh707e725_0    conda-forge
cloudpickle               3.1.1              pyhd8ed1ab_0    conda-forge
contourpy                 1.3.1           py313h33d0bda_0    conda-forge
cpython                   3.13.2          py313hd8ed1ab_100    conda-forge
cuda-crt-tools            12.8.61              ha770c72_1    conda-forge
cuda-cudart               12.8.57              h5888daf_1    conda-forge
cuda-cudart_linux-64      12.8.57              h3f2d84a_1    conda-forge
cuda-cuobjdump            12.8.55              hbd13f7d_0    conda-forge
cuda-cupti                12.8.57              hbd13f7d_0    conda-forge
cuda-nvcc-tools           12.8.61              he02047a_1    conda-forge
cuda-nvdisasm             12.8.55              hbd13f7d_0    conda-forge
cuda-nvrtc                12.8.61              hbd13f7d_0    conda-forge
cuda-nvtx                 12.8.55              hbd13f7d_0    conda-forge
cuda-nvvm-tools           12.8.61              he02047a_1    conda-forge
cuda-version              12.8                 h5d125a7_3    conda-forge
cudnn                     9.7.1.26             h969bcc4_0    conda-forge
cusparselt                0.7.0.0              hcd2ec93_0    conda-forge
cycler                    0.12.1             pyhd8ed1ab_1    conda-forge
dask-core                 2024.8.2           pyhd8ed1ab_0    conda-forge
datapi                    0.2.0                    pypi_0    pypi
earthkit-data             0.12.2                   pypi_0    pypi
earthkit-meteo            0.3.0                    pypi_0    pypi
earthkit-regrid           0.4.0                    pypi_0    pypi
eccodes                   2.39.0               hf413ef6_1    conda-forge
ecmwf-api-client          1.6.3                    pypi_0    pypi
ecmwf-opendata            0.3.14                   pypi_0    pypi
entrypoints               0.4                      pypi_0    pypi
filelock                  3.17.0             pyhd8ed1ab_0    conda-forge
findlibs                  0.0.5              pyhd8ed1ab_0    conda-forge
fonttools                 4.56.0          py313h8060acc_0    conda-forge
freeglut                  3.2.2                ha6d2627_3    conda-forge
freetype                  2.12.1               h267a509_2    conda-forge
fsspec                    2025.2.0           pyhd8ed1ab_0    conda-forge
geos                      3.13.0               h5888daf_0    conda-forge
gmp                       6.3.0                hac33072_2    conda-forge
gmpy2                     2.1.5           py313h11186cd_3    conda-forge
gputil                    1.4.0                    pypi_0    pypi
h2                        4.2.0              pyhd8ed1ab_0    conda-forge
hdf4                      4.2.15               h2a13503_7    conda-forge
hdf5                      1.14.4          nompi_h2d575fe_105    conda-forge
hpack                     4.1.0              pyhd8ed1ab_0    conda-forge
hyperframe                6.1.0              pyhd8ed1ab_0    conda-forge
idna                      3.10                     pypi_0    pypi
importlib-metadata        8.6.1              pyha770c72_0    conda-forge
importlib_metadata        8.6.1                hd8ed1ab_0    conda-forge
iris                      3.11.1             pyha770c72_1    conda-forge
iris-grib                 0.21.0             pyhd8ed1ab_0    conda-forge
jasper                    4.2.4                h536e39c_0    conda-forge
jinja2                    3.1.5              pyhd8ed1ab_0    conda-forge
jmespath                  1.0.1              pyhd8ed1ab_1    conda-forge
jsonschema                4.23.0                   pypi_0    pypi
jsonschema-specifications 2024.10.1                pypi_0    pypi
keyutils                  1.6.1                h166bdaf_0    conda-forge
kiwisolver                1.4.7           py313h33d0bda_0    conda-forge
krb5                      1.21.3               h659f571_0    conda-forge
lcms2                     2.17                 h717163a_0    conda-forge
ld_impl_linux-64          2.43                 h712a8e2_2    conda-forge
lerc                      4.0.0                h27087fc_0    conda-forge
libabseil                 20240722.0      cxx17_hbbce691_4    conda-forge
libaec                    1.1.3                h59595ed_0    conda-forge
libblas                   3.9.0           29_h2556b6b_mkl    conda-forge
libbrotlicommon           1.1.0                hb9d3cd8_2    conda-forge
libbrotlidec              1.1.0                hb9d3cd8_2    conda-forge
libbrotlienc              1.1.0                hb9d3cd8_2    conda-forge
libcap                    2.71                 h39aace5_0    conda-forge
libcblas                  3.9.0           29_h372d94f_mkl    conda-forge
libcublas                 12.8.3.14            h9ab20c4_0    conda-forge
libcudss0                 0.4.0.2              he55f5cd_2    conda-forge
libcufft                  11.3.3.41            hbd13f7d_0    conda-forge
libcufile                 1.13.0.11            h12f29b5_0    conda-forge
libcurand                 10.3.9.55            hbd13f7d_0    conda-forge
libcurl                   8.12.1               h332b0f4_0    conda-forge
libcusolver               11.7.2.55            h9ab20c4_0    conda-forge
libcusparse               12.5.7.53            hbd13f7d_0    conda-forge
libdeflate                1.23                 h4ddbbb0_0    conda-forge
libdrm                    2.4.124              hb9d3cd8_0    conda-forge
libedit                   3.1.20250104    pl5321h7949ede_0    conda-forge
libegl                    1.7.0                ha4b6fd6_2    conda-forge
libev                     4.33                 hd590300_2    conda-forge
libexpat                  2.6.4                h5888daf_0    conda-forge
libffi                    3.4.6                h2dba641_0    conda-forge
libgcc                    14.2.0               h77fa898_1    conda-forge
libgcc-ng                 14.2.0               h69a702a_1    conda-forge
libgcrypt-lib             1.11.0               hb9d3cd8_2    conda-forge
libgfortran               14.2.0               h69a702a_1    conda-forge
libgfortran5              14.2.0               hd5240d6_1    conda-forge
libgl                     1.7.0                ha4b6fd6_2    conda-forge
libglu                    9.0.3                h03adeef_0    conda-forge
libglvnd                  1.7.0                ha4b6fd6_2    conda-forge
libglx                    1.7.0                ha4b6fd6_2    conda-forge
libgpg-error              1.51                 hbd13f7d_1    conda-forge
libhwloc                  2.11.2          default_h0d58e46_1001    conda-forge
libiconv                  1.17                 hd590300_2    conda-forge
libjpeg-turbo             3.0.0                hd590300_1    conda-forge
liblapack                 3.9.0           29_hc41d3b0_mkl    conda-forge
libllvm20                 20.1.0.rc1           hb8a341e_0    conda-forge
liblzma                   5.6.4                hb9d3cd8_0    conda-forge
libmagma                  2.8.0                h566cb83_2    conda-forge
libmpdec                  4.0.0                h4bc722e_0    conda-forge
libnetcdf                 4.9.2           nompi_h5ddbaa4_116    conda-forge
libnghttp2                1.64.0               h161d5f1_0    conda-forge
libnl                     3.11.0               hb9d3cd8_0    conda-forge
libnvjitlink              12.8.61              hbd13f7d_0    conda-forge
libpciaccess              0.18                 hd590300_0    conda-forge
libpng                    1.6.46               h943b412_0    conda-forge
libprotobuf               5.28.3               h6128344_1    conda-forge
libsqlite                 3.48.0               hee588c1_1    conda-forge
libssh2                   1.11.1               hf672d98_0    conda-forge
libstdcxx                 14.2.0               hc0a3c3a_1    conda-forge
libstdcxx-ng              14.2.0               h4852527_1    conda-forge
libsystemd0               257.3                h3dc2cb9_0    conda-forge
libtiff                   4.7.0                hd9ff511_3    conda-forge
libtorch                  2.6.0           cuda126_mkl_h8247c52_300    conda-forge
libudev1                  257.3                h9a4d06a_0    conda-forge
libudunits2               2.2.28               h40f5838_3    conda-forge
libuuid                   2.38.1               h0b41bf4_0    conda-forge
libuv                     1.50.0               hb9d3cd8_0    conda-forge
libwebp-base              1.5.0                h851e524_0    conda-forge
libxcb                    1.17.0               h8a09558_0    conda-forge
libxml2                   2.13.5               h0d44e9d_1    conda-forge
libzip                    1.11.2               h6991a6a_0    conda-forge
libzlib                   1.3.1                hb9d3cd8_2    conda-forge
llvm-openmp               19.1.7               h024ca30_0    conda-forge
locket                    1.0.0              pyhd8ed1ab_0    conda-forge
lru-dict                  1.3.0                    pypi_0    pypi
lz4-c                     1.10.0               h5888daf_1    conda-forge
markdown                  3.7                      pypi_0    pypi
markupsafe                3.0.2           py313h8060acc_1    conda-forge
matplotlib-base           3.10.0          py313h129903b_0    conda-forge
mkl                       2024.2.2            ha957f24_16    conda-forge
mpc                       1.3.1                h24ddda3_1    conda-forge
mpfr                      4.2.1                h90cbb55_3    conda-forge
mpmath                    1.3.0              pyhd8ed1ab_1    conda-forge
multiurl                  0.3.3                    pypi_0    pypi
munkres                   1.1.4              pyh9f0ad1d_0    conda-forge
nccl                      2.25.1.1             ha44e49d_0    conda-forge
ncurses                   6.5                  h2d0b736_3    conda-forge
netcdf4                   1.7.2           nompi_py313h1dd084c_101    conda-forge
networkx                  3.4.2              pyh267e887_2    conda-forge
numpy                     1.26.4                   pypi_0    pypi
openjpeg                  2.5.3                h5fbd93e_0    conda-forge
openssl                   3.4.1                h7b32b05_0    conda-forge
optree                    0.14.0          py313h33d0bda_1    conda-forge
packaging                 24.2               pyhd8ed1ab_2    conda-forge
pandas                    2.2.3           py313ha87cce1_1    conda-forge
partd                     1.4.2              pyhd8ed1ab_0    conda-forge
pdbufr                    0.12.2                   pypi_0    pypi
pillow                    11.1.0          py313h8db990d_0    conda-forge
pip                       25.0.1             pyh145f28c_0    conda-forge
proj                      9.5.1                h0054346_0    conda-forge
pthread-stubs             0.4               hb9d3cd8_1002    conda-forge
pybind11                  2.13.6             pyh1ec8472_2    conda-forge
pybind11-global           2.13.6             pyh415d2e4_2    conda-forge
pycparser                 2.22               pyh29332c3_1    conda-forge
pygrib                    2.1.6           py313hb163c74_1    conda-forge
pyparsing                 3.2.1              pyhd8ed1ab_0    conda-forge
pyproj                    3.7.1           py313hdb96ca5_0    conda-forge
pyshp                     2.3.1              pyhd8ed1ab_1    conda-forge
pysocks                   1.7.1              pyha55dd90_7    conda-forge
python                    3.13.2          hf636f53_100_cp313    conda-forge
python-dateutil           2.9.0.post0        pyhff2d567_1    conda-forge
python-eccodes            2.37.0          py313ha014f3b_0    conda-forge
python-tzdata             2025.1             pyhd8ed1ab_0    conda-forge
python-xxhash             3.5.0           py313h536fd9c_1    conda-forge
python_abi                3.13                    5_cp313    conda-forge
pytorch                   2.6.0           cuda126_mkl_py313_h7373160_300    conda-forge
pytz                      2024.1             pyhd8ed1ab_0    conda-forge
pyyaml                    6.0.2           py313h8060acc_2    conda-forge
qhull                     2020.2               h434a139_5    conda-forge
rdma-core                 56.0                 h5888daf_0    conda-forge
readline                  8.2                  h8228510_1    conda-forge
referencing               0.36.2                   pypi_0    pypi
requests                  2.32.3                   pypi_0    pypi
rpds-py                   0.22.3                   pypi_0    pypi
ruamel-yaml               0.18.10                  pypi_0    pypi
s3transfer                0.11.2             pyhd8ed1ab_0    conda-forge
scipy                     1.15.1          py313h750cbce_0    conda-forge
setuptools                75.8.0             pyhff2d567_0    conda-forge
shapely                   2.0.7           py313h3f71f02_0    conda-forge
six                       1.17.0             pyhd8ed1ab_0    conda-forge
sleef                     3.8                  h1b44611_0    conda-forge
snappy                    1.2.1                h8bd8927_1    conda-forge
sqlite                    3.48.0               h9eae976_1    conda-forge
sympy                     1.13.3           pyh2585a3b_105    conda-forge
tbb                       2021.13.0            hceb3a55_1    conda-forge
tk                        8.6.13          noxft_h4845f30_101    conda-forge
toolz                     1.0.0              pyhd8ed1ab_1    conda-forge
torch-harmonics           0.6.2                    pypi_0    pypi
tqdm                      4.67.1                   pypi_0    pypi
triton                    3.2.0           cuda126py313h46f6bd1_0    conda-forge
typing-extensions         4.12.2               hd8ed1ab_1    conda-forge
typing_extensions         4.12.2             pyha770c72_1    conda-forge
tzdata                    2025a                h78e105d_0    conda-forge
udunits2                  2.2.28               h40f5838_3    conda-forge
urllib3                   2.3.0              pyhd8ed1ab_0    conda-forge
xarray                    2025.1.2           pyhd8ed1ab_0    conda-forge
xorg-libx11               1.8.11               h4f16b4b_0    conda-forge
xorg-libxau               1.0.12               hb9d3cd8_0    conda-forge
xorg-libxdamage           1.1.6                hb9d3cd8_0    conda-forge
xorg-libxdmcp             1.1.5                hb9d3cd8_0    conda-forge
xorg-libxext              1.3.6                hb9d3cd8_0    conda-forge
xorg-libxfixes            6.0.1                hb9d3cd8_0    conda-forge
xorg-libxi                1.8.2                hb9d3cd8_0    conda-forge
xorg-libxxf86vm           1.1.6                hb9d3cd8_0    conda-forge
xxhash                    0.8.2                hd590300_0    conda-forge
yaml                      0.2.5                h7f98852_2    conda-forge
zipp                      3.21.0             pyhd8ed1ab_1    conda-forge
zlib                      1.3.1                hb9d3cd8_2    conda-forge
zstandard                 0.23.0          py313h80202fe_1    conda-forge
zstd                      1.5.6                ha6fb4c9_0    conda-forge

(fourcastnet) jupyter@instance-20250216-164054:~/FourCastNet/NCEP/model$ pip freeze
ai-models==0.7.4
ai-models-fourcastnetv2==0.0.3
array_api_compat==1.10.0
attrs @ file:///home/conda/feedstock_root/build_artifacts/attrs_1737819173731/work
boto3 @ file:///home/conda/feedstock_root/build_artifacts/boto3_1739609553469/work
botocore @ file:///home/conda/feedstock_root/build_artifacts/botocore_1739607462647/work
Brotli @ file:///home/conda/feedstock_root/build_artifacts/brotli-split_1725267488082/work
Cartopy @ file:///home/conda/feedstock_root/build_artifacts/cartopy_1728342222487/work
cdsapi==0.7.5
certifi @ file:///home/conda/feedstock_root/build_artifacts/certifi_1739515848642/work/certifi
cf-units @ file:///home/conda/feedstock_root/build_artifacts/cf-units_1730445509006/work
cffi @ file:///home/conda/feedstock_root/build_artifacts/cffi_1725560556141/work
cfgrib==0.9.15.0
cftime @ file:///home/conda/feedstock_root/build_artifacts/cftime_1725400458563/work
cfunits @ file:///home/conda/feedstock_root/build_artifacts/cfunits_1734958285439/work
charset-normalizer==3.4.1
click @ file:///home/conda/feedstock_root/build_artifacts/click_1734858813237/work
cloudpickle @ file:///home/conda/feedstock_root/build_artifacts/cloudpickle_1736947526808/work
contourpy @ file:///home/conda/feedstock_root/build_artifacts/contourpy_1731428302318/work
cycler @ file:///home/conda/feedstock_root/build_artifacts/cycler_1733332471406/work
dask @ file:///home/conda/feedstock_root/build_artifacts/dask-core_1725051073088/work
datapi==0.2.0
earthkit-data==0.12.2
earthkit-meteo==0.3.0
earthkit-regrid==0.4.0
eccodes @ file:///home/conda/feedstock_root/build_artifacts/python-eccodes_1725958396792/work
ecmwf-api-client==1.6.3
ecmwf-opendata==0.3.14
entrypoints==0.4
filelock @ file:///home/conda/feedstock_root/build_artifacts/filelock_1737517818712/work
findlibs @ file:///home/conda/feedstock_root/build_artifacts/findlibs_1682423883580/work
fonttools @ file:///home/conda/feedstock_root/build_artifacts/fonttools_1738940303262/work
fsspec @ file:///home/conda/feedstock_root/build_artifacts/fsspec_1738501244951/work
gmpy2 @ file:///home/conda/feedstock_root/build_artifacts/gmpy2_1733462551614/work
GPUtil==1.4.0
h2 @ file:///home/conda/feedstock_root/build_artifacts/h2_1738578511449/work
hpack @ file:///home/conda/feedstock_root/build_artifacts/hpack_1737618293087/work
hyperframe @ file:///home/conda/feedstock_root/build_artifacts/hyperframe_1737618333194/work
idna==3.10
importlib_metadata @ file:///home/conda/feedstock_root/build_artifacts/importlib-metadata_1737420181517/work
iris-grib @ file:///home/conda/feedstock_root/build_artifacts/iris-grib_1736338015682/work
Jinja2 @ file:///home/conda/feedstock_root/build_artifacts/jinja2_1734823942230/work
jmespath @ file:///home/conda/feedstock_root/build_artifacts/jmespath_1733229141657/work
jsonschema==4.23.0
jsonschema-specifications==2024.10.1
kiwisolver @ file:///home/conda/feedstock_root/build_artifacts/kiwisolver_1725459255340/work
locket @ file:///home/conda/feedstock_root/build_artifacts/locket_1650660393415/work
lru-dict==1.3.0
Markdown==3.7
MarkupSafe @ file:///home/conda/feedstock_root/build_artifacts/markupsafe_1733219680183/work
matplotlib==3.10.0
mpmath @ file:///home/conda/feedstock_root/build_artifacts/mpmath_1733302684489/work
multiurl==0.3.3
munkres==1.1.4
netCDF4 @ file:///home/conda/feedstock_root/build_artifacts/netcdf4_1733253092579/work
networkx @ file:///home/conda/feedstock_root/build_artifacts/bld/rattler-build_networkx_1731521053/work
numpy==1.26.4
optree @ file:///home/conda/feedstock_root/build_artifacts/optree_1737742729671/work
packaging @ file:///home/conda/feedstock_root/build_artifacts/packaging_1733203243479/work
pandas @ file:///home/conda/feedstock_root/build_artifacts/pandas_1726878403859/work
partd @ file:///home/conda/feedstock_root/build_artifacts/partd_1715026491486/work
pdbufr==0.12.2
pillow @ file:///home/conda/feedstock_root/build_artifacts/pillow_1735929688808/work
pybind11 @ file:///D:/bld/pybind11-split_1730237329882/work
pybind11_global @ file:///home/conda/feedstock_root/build_artifacts/pybind11-split_1730237328107/work
pycparser @ file:///home/conda/feedstock_root/build_artifacts/bld/rattler-build_pycparser_1733195786/work
pygrib @ file:///home/conda/feedstock_root/build_artifacts/pygrib_1727366176096/work
pyparsing @ file:///home/conda/feedstock_root/build_artifacts/pyparsing_1735698276679/work
pyproj @ file:///home/conda/feedstock_root/build_artifacts/pyproj_1739711615313/work
pyshp @ file:///home/conda/feedstock_root/build_artifacts/pyshp_1733821528126/work
PySocks @ file:///home/conda/feedstock_root/build_artifacts/pysocks_1733217236728/work
python-dateutil @ file:///home/conda/feedstock_root/build_artifacts/python-dateutil_1733215673016/work
pytz @ file:///home/conda/feedstock_root/build_artifacts/pytz_1706886791323/work
PyYAML @ file:///home/conda/feedstock_root/build_artifacts/pyyaml_1737454647378/work
referencing==0.36.2
requests==2.32.3
rpds-py==0.22.3
ruamel.yaml==0.18.10
s3transfer @ file:///home/conda/feedstock_root/build_artifacts/s3transfer_1737737761316/work
scipy @ file:///home/conda/feedstock_root/build_artifacts/scipy-split_1736617218858/work/dist/scipy-1.15.1-cp313-cp313-linux_x86_64.whl#sha256=e91cc6d084ad66b8f58747e677f2f31d098df45e70ec382812235c7fd2cfdbbc
scitools-iris @ file:///home/conda/feedstock_root/build_artifacts/iris_1737530532875/work
setuptools==75.8.0
shapely @ file:///home/conda/feedstock_root/build_artifacts/shapely_1738307919858/work
six @ file:///home/conda/feedstock_root/build_artifacts/six_1733380938961/work
sympy @ file:///home/conda/feedstock_root/build_artifacts/sympy_1736248176451/work
toolz @ file:///home/conda/feedstock_root/build_artifacts/toolz_1733736030883/work
torch @ file:///home/conda/feedstock_root/build_artifacts/libtorch_1739474903285/work
torch-harmonics==0.6.2
tqdm==4.67.1
triton @ file:///home/conda/feedstock_root/build_artifacts/triton_1738673999158/work/python
typing_extensions @ file:///home/conda/feedstock_root/build_artifacts/typing_extensions_1733188668063/work
tzdata @ file:///home/conda/feedstock_root/build_artifacts/python-tzdata_1737541069190/work
urllib3 @ file:///home/conda/feedstock_root/build_artifacts/urllib3_1734859416348/work
xarray @ file:///home/conda/feedstock_root/build_artifacts/xarray_1738313659268/work
xxhash @ file:///home/conda/feedstock_root/build_artifacts/python-xxhash_1725272044304/work
zipp @ file:///home/conda/feedstock_root/build_artifacts/zipp_1732827521216/work
zstandard==0.23.0

@MrCsabaToth
Copy link
Author 

(fourcastnet) jupyter@instance-20250216-164054:~/FourCastNet/NCEP$ unzip -lt weights.tar
Archive:  weights.tar
    testing: weights/data.pkl         OK
    testing: weights/data/0           OK
    testing: weights/data/1           OK
    testing: weights/data/10          OK
    testing: weights/data/100         OK
    testing: weights/data/101         OK
    testing: weights/data/102         OK
    testing: weights/data/103         OK
    testing: weights/data/104         OK
    testing: weights/data/105         OK
    testing: weights/data/106         OK
    testing: weights/data/107         OK
    testing: weights/data/108         OK
    testing: weights/data/109         OK
    testing: weights/data/11          OK
    testing: weights/data/110         OK
    testing: weights/data/111         OK
    testing: weights/data/112         OK
    testing: weights/data/113         OK
    testing: weights/data/114         OK
    testing: weights/data/115         OK
    testing: weights/data/116         OK
    testing: weights/data/117         OK
    testing: weights/data/118         OK
    testing: weights/data/119         OK
    testing: weights/data/12          OK
    testing: weights/data/120         OK
    testing: weights/data/121         OK
    testing: weights/data/122         OK
    testing: weights/data/123         OK
    testing: weights/data/124         OK
    testing: weights/data/125         OK
    testing: weights/data/126         OK
    testing: weights/data/127         OK
    testing: weights/data/128         OK
    testing: weights/data/129         OK
    testing: weights/data/13          OK
    testing: weights/data/130         OK
    testing: weights/data/131         OK
    testing: weights/data/132         OK
    testing: weights/data/133         OK
    testing: weights/data/134         OK
    testing: weights/data/135         OK
    testing: weights/data/136         OK
    testing: weights/data/137         OK
    testing: weights/data/138         OK
    testing: weights/data/139         OK
    testing: weights/data/14          OK
    testing: weights/data/140         OK
    testing: weights/data/141         OK
    testing: weights/data/142         OK
    testing: weights/data/143         OK
    testing: weights/data/144         OK
    testing: weights/data/145         OK
    testing: weights/data/146         OK
    testing: weights/data/147         OK
    testing: weights/data/148         OK
    testing: weights/data/149         OK
    testing: weights/data/15          OK
    testing: weights/data/150         OK
    testing: weights/data/151         OK
    testing: weights/data/152         OK
    testing: weights/data/153         OK
    testing: weights/data/154         OK
    testing: weights/data/155         OK
    testing: weights/data/156         OK
    testing: weights/data/157         OK
    testing: weights/data/158         OK
    testing: weights/data/159         OK
    testing: weights/data/16          OK
    testing: weights/data/160         OK
    testing: weights/data/161         OK
    testing: weights/data/162         OK
    testing: weights/data/163         OK
    testing: weights/data/164         OK
    testing: weights/data/165         OK
    testing: weights/data/166         OK
    testing: weights/data/167         OK
    testing: weights/data/168         OK
    testing: weights/data/169         OK
    testing: weights/data/17          OK
    testing: weights/data/170         OK
    testing: weights/data/171         OK
    testing: weights/data/172         OK
    testing: weights/data/173         OK
    testing: weights/data/174         OK
    testing: weights/data/175         OK
    testing: weights/data/176         OK
    testing: weights/data/177         OK
    testing: weights/data/178         OK
    testing: weights/data/179         OK
    testing: weights/data/18          OK
    testing: weights/data/180         OK
    testing: weights/data/181         OK
    testing: weights/data/182         OK
    testing: weights/data/183         OK
    testing: weights/data/184         OK
    testing: weights/data/185         OK
    testing: weights/data/186         OK
    testing: weights/data/187         OK
    testing: weights/data/188         OK
    testing: weights/data/189         OK
    testing: weights/data/19          OK
    testing: weights/data/190         OK
    testing: weights/data/191         OK
    testing: weights/data/192         OK
    testing: weights/data/193         OK
    testing: weights/data/194         OK
    testing: weights/data/195         OK
    testing: weights/data/196         OK
    testing: weights/data/197         OK
    testing: weights/data/198         OK
    testing: weights/data/199         OK
    testing: weights/data/2           OK
    testing: weights/data/20          OK
    testing: weights/data/200         OK
    testing: weights/data/201         OK
    testing: weights/data/202         OK
    testing: weights/data/203         OK
    testing: weights/data/204         OK
    testing: weights/data/205         OK
    testing: weights/data/206         OK
    testing: weights/data/207         OK
    testing: weights/data/208         OK
    testing: weights/data/209         OK
    testing: weights/data/21          OK
    testing: weights/data/210         OK
    testing: weights/data/211         OK
    testing: weights/data/212         OK
    testing: weights/data/213         OK
    testing: weights/data/214         OK
    testing: weights/data/215         OK
    testing: weights/data/216         OK
    testing: weights/data/217         OK
    testing: weights/data/218         OK
    testing: weights/data/219         OK
    testing: weights/data/22          OK
    testing: weights/data/220         OK
    testing: weights/data/221         OK
    testing: weights/data/222         OK
    testing: weights/data/223         OK
    testing: weights/data/224         OK
    testing: weights/data/225         OK
    testing: weights/data/226         OK
    testing: weights/data/227         OK
    testing: weights/data/228         OK
    testing: weights/data/229         OK
    testing: weights/data/23          OK
    testing: weights/data/230         OK
    testing: weights/data/231         OK
    testing: weights/data/232         OK
    testing: weights/data/233         OK
    testing: weights/data/234         OK
    testing: weights/data/235         OK
    testing: weights/data/236         OK
    testing: weights/data/237         OK
    testing: weights/data/238         OK
    testing: weights/data/239         OK
    testing: weights/data/24          OK
    testing: weights/data/240         OK
    testing: weights/data/241         OK
    testing: weights/data/242         OK
    testing: weights/data/243         OK
    testing: weights/data/244         OK
    testing: weights/data/245         OK
    testing: weights/data/246         OK
    testing: weights/data/247         OK
    testing: weights/data/248         OK
    testing: weights/data/249         OK
    testing: weights/data/25          OK
    testing: weights/data/250         OK
    testing: weights/data/251         OK
    testing: weights/data/252         OK
    testing: weights/data/253         OK
    testing: weights/data/254         OK
    testing: weights/data/255         OK
    testing: weights/data/256         OK
    testing: weights/data/257         OK
    testing: weights/data/258         OK
    testing: weights/data/259         OK
    testing: weights/data/26          OK
    testing: weights/data/260         OK
    testing: weights/data/261         OK
    testing: weights/data/262         OK
    testing: weights/data/263         OK
    testing: weights/data/264         OK
    testing: weights/data/265         OK
    testing: weights/data/266         OK
    testing: weights/data/267         OK
    testing: weights/data/268         OK
    testing: weights/data/269         OK
    testing: weights/data/27          OK
    testing: weights/data/270         OK
    testing: weights/data/271         OK
    testing: weights/data/272         OK
    testing: weights/data/273         OK
    testing: weights/data/274         OK
    testing: weights/data/275         OK
    testing: weights/data/276         OK
    testing: weights/data/277         OK
    testing: weights/data/278         OK
    testing: weights/data/279         OK
    testing: weights/data/28          OK
    testing: weights/data/280         OK
    testing: weights/data/281         OK
    testing: weights/data/282         OK
    testing: weights/data/283         OK
    testing: weights/data/284         OK
    testing: weights/data/285         OK
    testing: weights/data/286         OK
    testing: weights/data/287         OK
    testing: weights/data/288         OK
    testing: weights/data/289         OK
    testing: weights/data/29          OK
    testing: weights/data/290         OK
    testing: weights/data/291         OK
    testing: weights/data/292         OK
    testing: weights/data/293         OK
    testing: weights/data/294         OK
    testing: weights/data/295         OK
    testing: weights/data/296         OK
    testing: weights/data/297         OK
    testing: weights/data/298         OK
    testing: weights/data/299         OK
    testing: weights/data/3           OK
    testing: weights/data/30          OK
    testing: weights/data/300         OK
    testing: weights/data/301         OK
    testing: weights/data/302         OK
    testing: weights/data/303         OK
    testing: weights/data/304         OK
    testing: weights/data/305         OK
    testing: weights/data/306         OK
    testing: weights/data/307         OK
    testing: weights/data/308         OK
    testing: weights/data/309         OK
    testing: weights/data/31          OK
    testing: weights/data/310         OK
    testing: weights/data/311         OK
    testing: weights/data/312         OK
    testing: weights/data/313         OK
    testing: weights/data/314         OK
    testing: weights/data/315         OK
    testing: weights/data/316         OK
    testing: weights/data/317         OK
    testing: weights/data/318         OK
    testing: weights/data/319         OK
    testing: weights/data/32          OK
    testing: weights/data/320         OK
    testing: weights/data/321         OK
    testing: weights/data/322         OK
    testing: weights/data/323         OK
    testing: weights/data/324         OK
    testing: weights/data/325         OK
    testing: weights/data/326         OK
    testing: weights/data/327         OK
    testing: weights/data/328         OK
    testing: weights/data/329         OK
    testing: weights/data/33          OK
    testing: weights/data/330         OK
    testing: weights/data/331         OK
    testing: weights/data/332         OK
    testing: weights/data/333         OK
    testing: weights/data/334         OK
    testing: weights/data/335         OK
    testing: weights/data/336         OK
    testing: weights/data/337         OK
    testing: weights/data/338         OK
    testing: weights/data/339         OK
    testing: weights/data/34          OK
    testing: weights/data/340         OK
    testing: weights/data/341         OK
    testing: weights/data/342         OK
    testing: weights/data/343         OK
    testing: weights/data/344         OK
    testing: weights/data/345         OK
    testing: weights/data/346         OK
    testing: weights/data/347         OK
    testing: weights/data/348         OK
    testing: weights/data/349         OK
    testing: weights/data/35          OK
    testing: weights/data/350         OK
    testing: weights/data/351         OK
    testing: weights/data/352         OK
    testing: weights/data/353         OK
    testing: weights/data/354         OK
    testing: weights/data/355         OK
    testing: weights/data/356         OK
    testing: weights/data/357         OK
    testing: weights/data/358         OK
    testing: weights/data/359         OK
    testing: weights/data/36          OK
    testing: weights/data/360         OK
    testing: weights/data/361         OK
    testing: weights/data/362         OK
    testing: weights/data/363         OK
    testing: weights/data/364         OK
    testing: weights/data/365         OK
    testing: weights/data/366         OK
    testing: weights/data/367         OK
    testing: weights/data/368         OK
    testing: weights/data/369         OK
    testing: weights/data/37          OK
    testing: weights/data/370         OK
    testing: weights/data/371         OK
    testing: weights/data/372         OK
    testing: weights/data/373         OK
    testing: weights/data/374         OK
    testing: weights/data/375         OK
    testing: weights/data/376         OK
    testing: weights/data/377         OK
    testing: weights/data/378         OK
    testing: weights/data/379         OK
    testing: weights/data/38          OK
    testing: weights/data/380         OK
    testing: weights/data/381         OK
    testing: weights/data/382         OK
    testing: weights/data/383         OK
    testing: weights/data/384         OK
    testing: weights/data/385         OK
    testing: weights/data/386         OK
    testing: weights/data/387         OK
    testing: weights/data/388         OK
    testing: weights/data/389         OK
    testing: weights/data/39          OK
    testing: weights/data/390         OK
    testing: weights/data/391         OK
    testing: weights/data/392         OK
    testing: weights/data/393         OK
    testing: weights/data/394         OK
    testing: weights/data/395         OK
    testing: weights/data/396         OK
    testing: weights/data/397         OK
    testing: weights/data/398         OK
    testing: weights/data/399         OK
    testing: weights/data/4           OK
    testing: weights/data/40          OK
    testing: weights/data/400         OK
    testing: weights/data/401         OK
    testing: weights/data/402         OK
    testing: weights/data/403         OK
    testing: weights/data/404         OK
    testing: weights/data/405         OK
    testing: weights/data/406         OK
    testing: weights/data/407         OK
    testing: weights/data/408         OK
    testing: weights/data/409         OK
    testing: weights/data/41          OK
    testing: weights/data/410         OK
    testing: weights/data/411         OK
    testing: weights/data/412         OK
    testing: weights/data/413         OK
    testing: weights/data/414         OK
    testing: weights/data/415         OK
    testing: weights/data/416         OK
    testing: weights/data/417         OK
    testing: weights/data/418         OK
    testing: weights/data/419         OK
    testing: weights/data/42          OK
    testing: weights/data/420         OK
    testing: weights/data/421         OK
    testing: weights/data/422         OK
    testing: weights/data/423         OK
    testing: weights/data/424         OK
    testing: weights/data/425         OK
    testing: weights/data/426         OK
    testing: weights/data/427         OK
    testing: weights/data/428         OK
    testing: weights/data/429         OK
    testing: weights/data/43          OK
    testing: weights/data/430         OK
    testing: weights/data/431         OK
    testing: weights/data/432         OK
    testing: weights/data/433         OK
    testing: weights/data/434         OK
    testing: weights/data/435         OK
    testing: weights/data/436         OK
    testing: weights/data/437         OK
    testing: weights/data/438         OK
    testing: weights/data/439         OK
    testing: weights/data/44          OK
    testing: weights/data/440         OK
    testing: weights/data/441         OK
    testing: weights/data/442         OK
    testing: weights/data/443         OK
    testing: weights/data/444         OK
    testing: weights/data/445         OK
    testing: weights/data/446         OK
    testing: weights/data/447         OK
    testing: weights/data/448         OK
    testing: weights/data/449         OK
    testing: weights/data/45          OK
    testing: weights/data/450         OK
    testing: weights/data/451         OK
    testing: weights/data/452         OK
    testing: weights/data/453         OK
    testing: weights/data/454         OK
    testing: weights/data/455         OK
    testing: weights/data/456         OK
    testing: weights/data/457         OK
    testing: weights/data/458         OK
    testing: weights/data/459         OK
    testing: weights/data/46          OK
    testing: weights/data/460         OK
    testing: weights/data/461         OK
    testing: weights/data/462         OK
    testing: weights/data/463         OK
    testing: weights/data/464         OK
    testing: weights/data/465         OK
    testing: weights/data/466         OK
    testing: weights/data/467         OK
    testing: weights/data/468         OK
    testing: weights/data/469         OK
    testing: weights/data/47          OK
    testing: weights/data/470         OK
    testing: weights/data/471         OK
    testing: weights/data/472         OK
    testing: weights/data/473         OK
    testing: weights/data/474         OK
    testing: weights/data/475         OK
    testing: weights/data/476         OK
    testing: weights/data/477         OK
    testing: weights/data/478         OK
    testing: weights/data/479         OK
    testing: weights/data/48          OK
    testing: weights/data/480         OK
    testing: weights/data/481         OK
    testing: weights/data/482         OK
    testing: weights/data/483         OK
    testing: weights/data/484         OK
    testing: weights/data/485         OK
    testing: weights/data/486         OK
    testing: weights/data/487         OK
    testing: weights/data/488         OK
    testing: weights/data/489         OK
    testing: weights/data/49          OK
    testing: weights/data/490         OK
    testing: weights/data/491         OK
    testing: weights/data/492         OK
    testing: weights/data/493         OK
    testing: weights/data/494         OK
    testing: weights/data/495         OK
    testing: weights/data/496         OK
    testing: weights/data/497         OK
    testing: weights/data/498         OK
    testing: weights/data/499         OK
    testing: weights/data/5           OK
    testing: weights/data/50          OK
    testing: weights/data/500         OK
    testing: weights/data/501         OK
    testing: weights/data/502         OK
    testing: weights/data/503         OK
    testing: weights/data/504         OK
    testing: weights/data/505         OK
    testing: weights/data/506         OK
    testing: weights/data/507         OK
    testing: weights/data/508         OK
    testing: weights/data/509         OK
    testing: weights/data/51          OK
    testing: weights/data/510         OK
    testing: weights/data/511         OK
    testing: weights/data/512         OK
    testing: weights/data/52          OK
    testing: weights/data/53          OK
    testing: weights/data/54          OK
    testing: weights/data/55          OK
    testing: weights/data/56          OK
    testing: weights/data/57          OK
    testing: weights/data/58          OK
    testing: weights/data/59          OK
    testing: weights/data/6           OK
    testing: weights/data/60          OK
    testing: weights/data/61          OK
    testing: weights/data/62          OK
    testing: weights/data/63          OK
    testing: weights/data/64          OK
    testing: weights/data/65          OK
    testing: weights/data/66          OK
    testing: weights/data/67          OK
    testing: weights/data/68          OK
    testing: weights/data/69          OK
    testing: weights/data/7           OK
    testing: weights/data/70          OK
    testing: weights/data/71          OK
    testing: weights/data/72          OK
    testing: weights/data/73          OK
    testing: weights/data/74          OK
    testing: weights/data/75          OK
    testing: weights/data/76          OK
    testing: weights/data/77          OK
    testing: weights/data/78          OK
    testing: weights/data/79          OK
    testing: weights/data/8           OK
    testing: weights/data/80          OK
    testing: weights/data/81          OK
    testing: weights/data/82          OK
    testing: weights/data/83          OK
    testing: weights/data/84          OK
    testing: weights/data/85          OK
    testing: weights/data/86          OK
    testing: weights/data/87          OK
    testing: weights/data/88          OK
    testing: weights/data/89          OK
    testing: weights/data/9           OK
    testing: weights/data/90          OK
    testing: weights/data/91          OK
    testing: weights/data/92          OK
    testing: weights/data/93          OK
    testing: weights/data/94          OK
    testing: weights/data/95          OK
    testing: weights/data/96          OK
    testing: weights/data/97          OK
    testing: weights/data/98          OK
    testing: weights/data/99          OK
    testing: weights/version          OK
No errors detected in compressed data of weights.tar.

@LinlinCui-NOAA
Copy link
Collaborator

@MrCsabaToth I am using torch 2.2.1. Here are torch related packages:

(mlwp) [Linlin.Cui@hfe07 realtime_forecast]$ conda list | grep torch
ffmpeg                    4.3                  hf484d3e_0    pytorch
pytorch                   2.2.1           py3.12_cuda12.1_cudnn8.9.2_0    pytorch
pytorch-cuda              12.1                 ha16c6d3_5    pytorch
pytorch-mutex             1.0                        cuda    pytorch
torch-harmonics           0.6.5                    pypi_0    pypi
torchaudio                2.2.1               py312_cu121    pytorch
torchvision               0.17.1              py312_cu121    pytorch
torchviz                  0.0.2                    pypi_0    pypi

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
2 participants
@MrCsabaToth @LinlinCui-NOAA