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:v2" install

To see all available flags, run the following command:

/usr/bin/docker run --rm "gcr.io/cos-cloud/cos-gpu-installer:v2" 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/build_and_release.sh <GCR-project> <image-tag>

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.