github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2025-07-21 02:41:25 +00:00
Code Issues Projects Releases Wiki Activity

Fix COS GPU driver installation

Browse Source 
* Rely on the built in GPU driver installer in COS as recommended in
  public docs - https://cloud.google.com/container-optimized-os/docs/how-to/run-gpus
* Run `nvidia-smi` after installation to verify installation
This commit is contained in:
David Porter 2021-10-28 17:49:50 -07:00
parent 8e37a3b324
commit e1a951afe5
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
test/e2e_node/jenkins/gci-init-gpu.yaml
View File

@ -2,15 +2,12 @@
runcmd:
- modprobe configs
# Setup the installation target at make it executable
- mkdir -p /home/kubernetes/bin/nvidia
- mount --bind /home/kubernetes/bin/nvidia /home/kubernetes/bin/nvidia
- mount -o remount,exec /home/kubernetes/bin/nvidia
# Compile and install the nvidia driver (precompiled driver installation currently fails)
- docker run --net=host --pid=host -v /dev:/dev -v /:/root -v /home/kubernetes/bin/nvidia:/usr/local/nvidia -e NVIDIA_INSTALL_DIR_HOST=/home/kubernetes/bin/nvidia -e NVIDIA_INSTALL_DIR_CONTAINER=/usr/local/nvidia -e NVIDIA_DRIVER_VERSION=460.91.03 --privileged gcr.io/cos-cloud/cos-gpu-installer:latest
# Run the installer again, as on the first try it doesn't detect the libnvidia-ml.so
# on the second attempt we detect it and update the ld cache.
- docker run --net=host --pid=host -v /dev:/dev -v /:/root -v /home/kubernetes/bin/nvidia:/usr/local/nvidia -e NVIDIA_INSTALL_DIR_HOST=/home/kubernetes/bin/nvidia -e NVIDIA_INSTALL_DIR_CONTAINER=/usr/local/nvidia -e NVIDIA_DRIVER_VERSION=460.91.03 --privileged gcr.io/cos-cloud/cos-gpu-installer:latest
# Install GPU drivers - https://cloud.google.com/container-optimized-os/docs/how-to/run-gpus
- cos-extensions install gpu
- mount --bind /var/lib/nvidia /var/lib/nvidia
- mount -o remount,exec /var/lib/nvidia /var/lib/nvidia
# Run nvidia-smi to verify installation
- /var/lib/nvidia/bin/nvidia-smi
# Remove build containers. They're very large.
- docker rm -f $(docker ps -aq)
# Standard installation proceeds