Update suite2p.smk to use cuda/12.4

Author: sumiya-kuroda

photon_mosaic/cli.py

@@ -310,6 +310,7 @@ def build_snakemake_command(args, config_path):
 
     cmd = [
         "snakemake",
+        "--use-envmodules", # Enables envmodules
         "--snakefile",
         str(snakefile_path),
         "--jobs",

photon_mosaic/workflow/suite2p.smk

@@ -63,6 +63,8 @@ rule suite2p:
                               for session_idx in discoverer.tiff_files[discoverer.original_datasets[i]].keys()]),
     resources:
         **(slurm_config if config.get("use_slurm") else {}),
+    envmodules: 
+        "cuda/12.4"
     run:
         from photon_mosaic.rules.suite2p_run import run_suite2p
         from pathlib import Path

pyproject.toml

@@ -12,7 +12,7 @@ dependencies = [
   "snakemake-executor-plugin-slurm",
   "PyYAML",
   "derotation",
-  "suite2p-NIU"
+  "suite2p-NIU>=0.1.1"
 ]
 
 license = {text = "BSD-3-Clause"}