Skip to content

Commit 06034ee

Browse files
authored
Added docker GPU instructions (#56)
1 parent fb29199 commit 06034ee

File tree

1 file changed

+53
-0
lines changed

1 file changed

+53
-0
lines changed

docs/installation.md

Lines changed: 53 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,3 +39,56 @@ You can also use [docker](https://hub.docker.com/r/giswqs/segment-geospatial/) t
3939
```bash
4040
docker run -it -p 8888:8888 giswqs/segment-geospatial:latest
4141
```
42+
43+
To enable GUI, run the following command to run a short benchmark on your GPU:
44+
45+
```bash
46+
docker run --rm -it --gpus=all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark
47+
```
48+
49+
The output should be similar to the following:
50+
51+
```text
52+
Run "nbody -benchmark [-numbodies=<numBodies>]" to measure performance.
53+
-fullscreen (run n-body simulation in fullscreen mode)
54+
-fp64 (use double precision floating point values for simulation)
55+
-hostmem (stores simulation data in host memory)
56+
-benchmark (run benchmark to measure performance)
57+
-numbodies=<N> (number of bodies (>= 1) to run in simulation)
58+
-device=<d> (where d=0,1,2.... for the CUDA device to use)
59+
-numdevices=<i> (where i=(number of CUDA devices > 0) to use for simulation)
60+
-compare (compares simulation results running once on the default GPU and once on the CPU)
61+
-cpu (run n-body simulation on the CPU)
62+
-tipsy=<file.bin> (load a tipsy model file for simulation)
63+
64+
NOTE: The CUDA Samples are not meant for performance measurements. Results may vary when GPU Boost is enabled.
65+
66+
> Windowed mode
67+
> Simulation data stored in video memory
68+
> Single precision floating point simulation
69+
> 1 Devices used for simulation
70+
GPU Device 0: "Turing" with compute capability 7.5
71+
72+
> Compute 7.5 CUDA device: [Quadro RTX 5000]
73+
49152 bodies, total time for 10 iterations: 69.386 ms
74+
= 348.185 billion interactions per second
75+
= 6963.703 single-precision GFLOP/s at 20 flops per interaction
76+
```
77+
78+
If you encounter the following error:
79+
80+
```text
81+
nvidia-container-cli: initialization error: load library failed: libnvidia-ml.so.1: cannot open shared object file: no such file or directory: unknown.
82+
```
83+
84+
Try adding `sudo` to the command:
85+
86+
```bash
87+
sudo docker run --rm -it --gpus=all nvcr.io/nvidia/k8s/cuda-sample:nbody nbody -gpu -benchmark
88+
```
89+
90+
Once everything is working, you can run the following command to start a Jupyter Notebook server:
91+
92+
```bash
93+
docker run -it -p 8888:8888 --gpus=all giswqs/segment-geospatial:latest
94+
```

0 commit comments

Comments
 (0)