-
Notifications
You must be signed in to change notification settings - Fork 1.4k
Closed
Description
with pytorch 1.9.x:
[2023-04-27T23:35:59.010Z] ======================================================================
[2023-04-27T23:35:59.010Z] ERROR: test_move_bundle_gen_folder (tests.test_auto3dseg_bundlegen.TestBundleGen)
[2023-04-27T23:35:59.010Z] ----------------------------------------------------------------------
[2023-04-27T23:35:59.010Z] Traceback (most recent call last):
[2023-04-27T23:35:59.010Z] File "/home/jenkins/agent/workspace/Monai-pytorch-versions/tests/test_auto3dseg_bundlegen.py", line 130, in test_move_bundle_gen_folder
[2023-04-27T23:35:59.010Z] bundle_generator.generate(work_dir, num_fold=1)
[2023-04-27T23:35:59.010Z] File "/home/jenkins/agent/workspace/Monai-pytorch-versions/monai/apps/auto3dseg/bundle_gen.py", line 589, in generate
[2023-04-27T23:35:59.010Z] gen_algo.export_to_disk(output_folder, name, fold=f_id)
[2023-04-27T23:35:59.010Z] File "/home/jenkins/agent/workspace/Monai-pytorch-versions/monai/apps/auto3dseg/bundle_gen.py", line 161, in export_to_disk
[2023-04-27T23:35:59.010Z] self.fill_records = self.fill_template_config(self.data_stats_files, self.output_path, **kwargs)
[2023-04-27T23:35:59.010Z] File "/tmp/tmpma08jta_/workdir/algorithm_templates/dints/scripts/algo.py", line 119, in fill_template_config
[2023-04-27T23:35:59.010Z] File "/tmp/tmpma08jta_/workdir/algorithm_templates/dints/scripts/algo.py", line 31, in get_mem_from_visible_gpus
[2023-04-27T23:35:59.010Z] AttributeError: module 'torch.cuda' has no attribute 'mem_get_info'
[2023-04-27T23:35:59.010Z]
[2023-04-27T23:35:59.010Z] ----------------------------------------------------------------------
Metadata
Metadata
Assignees
Labels
No labels