File tree Expand file tree Collapse file tree 1 file changed +36
-0
lines changed Expand file tree Collapse file tree 1 file changed +36
-0
lines changed Original file line number Diff line number Diff line change
1
+ #! /bin/bash
2
+
3
+ # Variable set in craype_config
4
+ JUHPC_SETUP_INSTALLDIR=$SCRATCH /${HOSTNAME%% -* } /juhpc_setup # HPC setup installation environment variables must be expanded during installation.
5
+
6
+ # Load required modules (including correct CPU and GPU target modules)
7
+ module load cray
8
+ module switch PrgEnv-cray PrgEnv-gnu
9
+ module load cudatoolkit craype-accel-nvidia90
10
+ module load cray-hdf5-parallel
11
+ module list
12
+
13
+ # Activate the HPC setup environment variables
14
+ . $JUHPC_SETUP_INSTALLDIR /activate
15
+
16
+ # Call juliaup to install juliaup and latest julia on scratch
17
+ juliaup
18
+
19
+ # Call juliaup to see its options
20
+ juliaup
21
+
22
+ # Call julia Pkg
23
+ julia -e ' using Pkg; Pkg.status()'
24
+
25
+ # Add CUDA.jl
26
+ julia -e ' using Pkg; Pkg.add("CUDA"); using CUDA; CUDA.versioninfo()'
27
+
28
+ # Add MPI.jl
29
+ julia -e ' using Pkg; Pkg.add("MPI"); using MPI; MPI.versioninfo()'
30
+
31
+ # Add HDF5.jl
32
+ julia -e ' using Pkg; Pkg.add("HDF5"); using HDF5; @show HDF5.has_parallel()'
33
+
34
+ # Test CUDA-aware MPI
35
+ cd ~ /cudaaware
36
+ MPICH_GPU_SUPPORT_ENABLED=1 srun -Acsstaff -C' gpu' -N2 -n2 julia cudaaware.jl
You can’t perform that action at this time.
0 commit comments