Skip to content

Commit 9dec50f

Browse files
authored
Merge pull request #3911 from zm711/nvidia_deps
Add in missing `get_nvidia_docker_dependencies`
2 parents 8c72f55 + 4c73f29 commit 9dec50f

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

src/spikeinterface/sorters/runsorter.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -479,8 +479,10 @@ def run_sorter_container(
479479
extra_kwargs["container_requires_gpu"] = True
480480

481481
if platform.system() == "Linux" and not has_docker_nvidia_installed():
482+
from .utils import get_nvidia_docker_dependencies
483+
482484
warn(
483-
f"nvidia-required but none of \n{get_nvidia_docker_dependecies()}\n were found. "
485+
f"nvidia-required but none of \n{get_nvidia_docker_dependencies()}\n were found. "
484486
f"This may result in an error being raised during sorting. Try "
485487
"installing `nvidia-container-toolkit`, including setting the "
486488
"configuration steps, if running into errors."

0 commit comments

Comments
 (0)