Skip to content
This repository was archived by the owner on Jan 11, 2023. It is now read-only.

Commit e929bb5

Browse files
committed
update GPU doc
1 parent f2c074b commit e929bb5

File tree

1 file changed

+5
-19
lines changed

1 file changed

+5
-19
lines changed

docs/kubernetes/gpu.md

+5-19
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ If `alpha.kubernetes.io/nvidia-gpu` is `0` and you just created the cluster, you
1919
When running a GPU container, you will need to specify how many GPU you want to use. If you don't specify a GPU count, kubernetes will asumme you don't require any, and will not map the device into the container.
2020
You will also need to mount the drivers from the host (the kubernetes agent) into the container.
2121

22-
On the host, the drivers are installed under `/usr/lib/nvidia-384`.
22+
On the host, the drivers are installed under `/usr/local/nvidia`.
2323

2424
Here is an example template running TensorFlow:
2525

@@ -45,27 +45,13 @@ spec:
4545
limits:
4646
alpha.kubernetes.io/nvidia-gpu: 1
4747
volumeMounts:
48-
- mountPath: /usr/local/nvidia/bin
49-
name: bin
50-
- mountPath: /usr/local/nvidia/lib64
51-
name: lib
52-
- mountPath: /usr/lib/x86_64-linux-gnu/libcuda.so.1
53-
name: libcuda
48+
- name: nvidia
49+
mountPath: /usr/local/nvidia
5450
volumes:
55-
- name: bin
51+
- name: nvidia
5652
hostPath:
57-
path: /usr/lib/nvidia-384/bin
58-
- name: lib
59-
hostPath:
60-
path: /usr/lib/nvidia-384
61-
- name: libcuda
62-
hostPath:
63-
path: /usr/lib/x86_64-linux-gnu/libcuda.so.1
53+
path: /usr/local/nvidia
6454
```
6555

6656
We specify `alpha.kubernetes.io/nvidia-gpu: 1` in the resources limits, and we mount the drivers from the host into the container.
6757

68-
Some libraries, such as `libcuda.so` are installed under `/usr/lib/x86_64-linux-gnu` on the host, you might need to mount them separatly as shown above based on your needs.
69-
70-
71-

0 commit comments

Comments
 (0)