tree: 4adf6a3b00fac4e7f1d5ea618f41dfa30a3067d3 [path history] [tgz]
  1. deviceinfo/
  2. features/
  3. internal/
  4. test/
  5. versions/
  6. cloudbuild.yaml
  7. Dockerfile
  8. main.go

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.


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 \
    "<tag>" install "-host-dir=/var/lib/nvidia"

To see all available flags, run the following command:

/usr/bin/docker run --rm "<tag>" help


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/ to run the test.