Skip to content

Commit ed072b8

Browse files
committed
Trying to get numba example to work
1 parent 115f5c0 commit ed072b8

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

docs/day4/gpu.rst

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -325,8 +325,7 @@ As before, we need a batch script to run the code. There are no GPUs on the logi
325325
salloc: Granted job allocation 32126787
326326
salloc: Waiting for resource configuration
327327
salloc: Nodes b-cn1606 are ready for job
328-
$ module load GCC/12.3.0 Python/3.11.3 OpenMPI/4.1.5 SciPy-bundle/2023.07 numba/0.58.1
329-
$ module load CUDA/12.1.1
328+
$ module load GCC/12.3.0 Python/3.11.3 OpenMPI/4.1.5 SciPy-bundle/2023.07 CUDA/12.1.1 numba/0.58.1
330329
$ srun python add-list.py
331330
CPU function took 14.216318 seconds.
332331
GPU function took 0.390335 seconds.
@@ -351,7 +350,6 @@ As before, we need a batch script to run the code. There are no GPUs on the logi
351350
# Remove any loaded modules and load the ones we need
352351
module purge > /dev/null 2>&1
353352
module load GCC/12.3.0 Python/3.11.3 OpenMPI/4.1.5 SciPy-bundle/2023.07 CUDA/12.1.1 numba/0.58.1
354-
module load CUDA/12.1.1
355353
356354
# Run your Python script
357355
python add-list.py

0 commit comments

Comments
 (0)