Skip to content

Commit 241c290

Browse files
committed
replace /tmp/juhpc with ./juhpc
1 parent 02beb2e commit 241c290

File tree

1 file changed

+7
-14
lines changed

1 file changed

+7
-14
lines changed

examples/cscs/daint/gpu/craype_config_no_cudaaware

Lines changed: 7 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -10,36 +10,29 @@
1010
# Load required modules, including correct CPU and GPU target modules (NOTE: the same modules should be loaded when running julia - JUHPC can be used in a module build recipe...)
1111
module load daint-gpu
1212
module switch PrgEnv-cray PrgEnv-gnu
13-
module load cudatoolkit # A toolkit needs to be loaded during installation in order to be able to set JUHPC_CUDA_HOME (even though we set preferences to use artifacts in the postinstall)
1413
module load cray-hdf5-parallel
1514
module list
1615

1716

1817
# Environment variables for HPC key packages that require system libraries that require system libraries (MPI.jl, CUDA.jl, AMDGPU.jl, HDF5.jl and ADIOS2.jl)
19-
export JUHPC_CUDA_HOME=$CUDA_HOME
20-
export JUHPC_CUDA_RUNTIME_VERSION="11.8" # Set to the highest possible version that can work on daint (using artifacts - below "local" is set to false...)
2118
export JUHPC_MPI_HOME=$MPICH_DIR
2219
export JUHPC_MPI_EXEC="srun -C gpu"
2320
export JUHPC_HDF5_HOME=$HDF5_DIR
2421

25-
2622
# Create site-specific post-install script (currently MPIPreferences does not provide an option to set required preloads if not automatically detected; JUHPC_MPI_VENDOR fails on Piz Daint...)
2723
JUHPC_POST_INSTALL_JL=./post_install.jl
28-
echo 'using Preferences
24+
echo 'using Pkg; Pkg.add("CUDA_Runtime_jll")
25+
using Preferences
2926
set_preferences!("CUDA_Runtime_jll",
27+
"version" => "11.8",
3028
"local" => false;
3129
force=true
32-
)
33-
set_preferences!("MPIPreferences",
34-
"preloads" => ["libcuda.so", "libcudart.so"],
35-
"preloads_env_switch" => "MPICH_RDMA_ENABLED_CUDA";
36-
force=true
37-
)' > $JUHPC_POST_INSTALL_JL
30+
)' > $JUHPC_POST_INSTALL_JL # (Set to the highest possible CUDA runtime version that can work on daint using artifacts - "local" is set to false...)
3831

3932

4033
# Call JUHPC
4134
JUHPC_SETUP_INSTALLDIR=$SCRATCH/../julia/${HOSTNAME%%[0-9]*}-gpu-nocudaaware/juhpc_setup
4235
JULIAUP_INSTALLDIR="\$SCRATCH/../julia/\$USER/\${HOSTNAME%%[0-9]*}-gpu-nocudaaware/juliaup"
43-
VERSION="v0.2.0"
44-
wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O /tmp/juhpc
45-
bash -l /tmp/juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR $JUHPC_POST_INSTALL_JL
36+
VERSION="v0.3.0"
37+
wget https://raw.githubusercontent.com/JuliaParallel/JUHPC/$VERSION/juhpc -O ./juhpc
38+
bash -l ./juhpc $JUHPC_SETUP_INSTALLDIR $JULIAUP_INSTALLDIR --postinstall=$JUHPC_POST_INSTALL_JL --verbose=1

0 commit comments

Comments
 (0)