tree: 1c1d3135f3a4d00a26a1a58b8fe20c166939e941 [path history] [tgz]
  1. internal/
  2. release/
  3. test/
  4. versions/
  5. cloudbuild.yaml
  6. Dockerfile
  7. 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 \
    "" install

To see all available flags, run the following command:

/usr/bin/docker run --rm "" help

Build and Release

Run the following script to build a cos_gpu_installer container image through Google Cloud Build and save the image to GCR:

src/cmd/cos_gpu_installer/release/ <GCR-project> <image-tag>


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.