Differences

This shows you the differences between two versions of the page.

--- techstaff:slurm [2016/05/09 14:50] – [sbatch] kauffman
+++ techstaff:slurm [2016/05/11 15:22] – [Using the GPU] kauffman
@@ Line 227: / Line 227: @@
 | error: Unable to allocate resources: More processors requested than permitted | It usually has **nothing** to do with priviledges you may or may not have. Rather, it usually means that you have allocated more processors than one compute node actually has. |
+====== Using the GPU ======
+===== Paths =====
+You will need to add the following to your $PATH and $LD_LIBRARY_PATH.
+  export PATH=$PATH:/usr/local/cuda/bin
+  export LD_LIBRARY_PATH=$LD_LIBRARY_PATH=/usr/local/cuda/lib
+===== Example =====
+This sbatch script will get device information from the installed Tesla gpu.
+<code>
+#!/bin/bash
+#
+#SBATCH --mail-user=cnetid@cs.uchicago.edu
+#SBATCH --mail-type=ALL
+#SBATCH --output=/home/cnetid/slurm/slurm_out/%j.%N.stdout
+#SBATCH --error=/home/cnetid/slurm/slurm_out/%j.%N.stderr
+#SBATCH --workdir=/home/cnetid/slurm
+#SBATCH --partition=gpu
+#SBATCH --job-name=get_tesla_info
+cat << EOF > /tmp/getinfo.cu
+#include <stdio.h>
+int main() {
+  int nDevices;
+  cudaGetDeviceCount(&nDevices);
+  for (int i = 0; i < nDevices; i++) {
+    cudaDeviceProp prop;
+    cudaGetDeviceProperties(&prop, i);
+    printf("Device Number: %d\n", i);
+    printf("  Device name: %s\n", prop.name);
+    printf("  Memory Clock Rate (KHz): %d\n",
+           prop.memoryClockRate);
+    printf("  Memory Bus Width (bits): %d\n",
+           prop.memoryBusWidth);
+    printf("  Peak Memory Bandwidth (GB/s): %f\n\n",
+.0*prop.memoryClockRate*(prop.memoryBusWidth/8)/1.0e6);
+  }
+}
+EOF
+/usr/local/cuda/bin/nvcc /tmp/getinfo.cu -o /tmp/a.out
+/tmp/a.out
+rm /tmp/a.out
+rm /tmp/getinfo.cu
+</code>
+==== Output ====
+STDOUT will look something like this:
+<code>
+cnetid@linux1:~$ cat $HOME/slurm/slurm_out/12567.gpu1.stdout
+Device Number: 0
+  Device name: Tesla M2090
+  Memory Clock Rate (KHz): 1848000
+  Memory Bus Width (bits): 384
+  Peak Memory Bandwidth (GB/s): 177.408000
+</code>
+STDERR should be blank.
 ====== More ======
 If you feel this documentation is lacking in some way please let techstaff know. Email [[techstaff@cs.uchicago.edu]], call (773-702-1031), or stop by our office (Ryerson 154).