Open mohawk811 opened 2 days ago
Hi @mohawk811, Have you tried the instructions given here to install DESC on Perlmutter? If not, please try that first instead of using a jax image.
Yes I followed those instructions and when I test the installation everything loads in fine
Ok, then use the following slurm script and you should be able to run a job.
#!/bin/bash
#SBATCH --qos=regular
#SBATCH --nodes=1
#SBATCH --ntasks-per-node=1
#SBATCH --constraint=gpu
#SBATCH --gpus 1
#SBATCH --mem=32G
#SBATCH --account=m4680
#SBATCH --time=03:25:00
export XLA_PYTHON_CLIENT_MEM_FRACTION=.93
XLA_PYTHON_CLIENT_MEM_FRACTION=.93
module load cudatoolkit/12.2;\
module load cudnn/8.9.3_cuda12;\
module load python;\
conda activate desc-env;\
python3 driver_ball_omni.py
Would it be possible if someone could share their slurm script for using a gpu for DESC on NERSC? I tried a number of things but this is the error I get from trying. I feel confident that my installation is fine from running the tests listed on the documentation.