tree: 28757d8d835992069651e2f66955ed01d64c3ec5 [path history] [tgz]
  1. deviceinfo/
  2. features/
  3. internal/
  4. test/
  5. versions/
  6. cloudbuild.yaml
  7. Dockerfile
  8. main.go
  9. README.md
src/cmd/cos_gpu_installer/README.md

COS GPU installer V2

cos_gpu_installer is a docker containers that can be used to download, compile and install GPU drivers on Container-Optimized OS images.

Usage

cos_gpu_installer should run on COS VM instances. Once you connect to your COS VM, run the following command to start a cos_gpu_installer container:

  /usr/bin/docker run --rm \
    --name="cos-gpu-installer" \
    --privileged \
    --net=host \
    --pid=host \
    --volume /dev:/dev \
    --volume /:/root \
    "gcr.io/cos-cloud/cos-gpu-installer:<tag>" install "-host-dir=/var/lib/nvidia"

To see all available flags, run the following command:

/usr/bin/docker run --rm "gcr.io/cos-cloud/cos-gpu-installer:<tag>" help

Test

Source code

Currently only unittest is available. Use go test to run unittest.

GPU drivers availability

The test test/check_drivers_test.go is available for checking GPU drivers availability. It checks which drivers are available for live COS images. Use test/run_test.sh to run the test.