test_swin_unetr

[wyli]

10:44:54  ======================================================================
10:44:54  ERROR: test_filter_swinunetr_0 (tests.test_swin_unetr.TestSWINUNETR)
10:44:54  ----------------------------------------------------------------------
10:44:54  Traceback (most recent call last):
10:44:54    File "/usr/local/lib/python3.10/dist-packages/parameterized/parameterized.py", line 620, in standalone_func
10:44:54      return func(*(a + p.args), **p.kwargs, **kw)
10:44:54    File "/home/jenkins/agent/workspace/MONAI-latest-docker-test/tests/test_swin_unetr.py", line 121, in test_filter_swinunetr
10:44:54      ssl_weight = torch.load(weight_path)["model"]
10:44:54    File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1016, in load
10:44:54      return _load(opened_zipfile,
10:44:54    File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1424, in _load
10:44:54      result = unpickler.load()
10:44:54    File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1394, in persistent_load
10:44:54      typed_storage = load_tensor(dtype, nbytes, key, _maybe_decode_ascii(location))
10:44:54    File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 1368, in load_tensor
10:44:54      wrap_storage=restore_location(storage, location),
10:44:54    File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 383, in default_restore_location
10:44:54      result = fn(storage, location)
10:44:54    File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 276, in _cuda_deserialize
10:44:54      device = validate_cuda_device(location)
10:44:54    File "/usr/local/lib/python3.10/dist-packages/torch/serialization.py", line 260, in validate_cuda_device
10:44:54      raise RuntimeError('Attempting to deserialize object on a CUDA '
10:44:54  RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.
10:44:54