Description
When I run brun_fastsurfer.sh with the tag '--parallel 3' using the most recent singularity container (including the singularity tags '--nv' and '--no-home'), I get the following error:
Traceback (most recent call last):
File "", line 1, in
ModuleNotFoundError: No module named 'torch.cuda.device_count'
I believe 'import torch.cuda.device_count as d' on line 257 of brun_fastsurfer.sh should be replaced by 'from torch.cuda import device_count as d', which worked when I ran the container interactively.
Steps to Reproduce
run brun_fastsurfer using latest fastsurfer singularity container with the tag '--parallel 3' (using singularity tags '--nv' and '--no-home')
Environment
singularity image built using the command: 'singularity build fastsurfer-gpu.sif docker://deepmi/fastsurfer'
singularity version 3.6.3
Description
When I run brun_fastsurfer.sh with the tag '--parallel 3' using the most recent singularity container (including the singularity tags '--nv' and '--no-home'), I get the following error:
Traceback (most recent call last):
File "", line 1, in
ModuleNotFoundError: No module named 'torch.cuda.device_count'
I believe 'import torch.cuda.device_count as d' on line 257 of brun_fastsurfer.sh should be replaced by 'from torch.cuda import device_count as d', which worked when I ran the container interactively.
Steps to Reproduce
run brun_fastsurfer using latest fastsurfer singularity container with the tag '--parallel 3' (using singularity tags '--nv' and '--no-home')
Environment
singularity image built using the command: 'singularity build fastsurfer-gpu.sif docker://deepmi/fastsurfer'
singularity version 3.6.3