init on gpu

CliMA · Jun 17, 2024 · 881a0cd · 881a0cd
1 parent 6839246
commit 881a0cd
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/.buildkite/longruns/pipeline.yml b/.buildkite/longruns/pipeline.yml
@@ -40,12 +40,12 @@ steps:
   #     JULIA_MAX_NUM_PRECOMPILE_FILES: 50
 
   - label: "init clima :computer:"
-    key: "init_cpu_env_clima"
+    key: "init_env_clima"
     command:
       - "echo $$JULIA_DEPOT_PATH"
 
       - echo "--- Instantiate AMIP env"
-      - "julia --project=experiments/ClimaEarth/ -e 'using Pkg; Pkg.develop(;path=\".\"); Pkg.add(url = \"https://github.com/CLiMA/RRTGMP.jl\", rev = \"zs/v0.14.0_checknan\");'"
+      - "julia --project=experiments/ClimaEarth/ -e 'using Pkg; Pkg.develop(;path=\".\"); Pkg.add(url = \"https://github.com/CliMA/RRTMGP.jl.git\", rev = \"zs/v0.14.0_checknan\");'"
       - "julia --project=experiments/ClimaEarth/ -e 'using Pkg; Pkg.instantiate(;verbose=true)'"
       - "julia --project=experiments/ClimaEarth/ -e 'using Pkg; Pkg.precompile()'"
       - "julia --project=experiments/ClimaEarth/ -e 'using Pkg; Pkg.status()'"
@@ -65,6 +65,8 @@ steps:
     agents:
       queue: clima
       modules: common
+      slurm_gpus: 1
+      slurm_cpus_per_task: 8
     env:
       JULIA_NUM_PRECOMPILE_TASKS: 8
       JULIA_MAX_NUM_PRECOMPILE_FILES: 50