Skip to content
This repository was archived by the owner on Jan 12, 2023. It is now read-only.
This repository was archived by the owner on Jan 12, 2023. It is now read-only.

gcr.io/cos-cloud/cos-gpu-installer:latest fails to install GPU drivers on compute instances #58

Description

@DanClarkeBuilds

Using gcr.io/cos-cloud/cos-gpu-installer:latest to install Nvidia GPU drivers on compute instances on start up fails to install drivers, as a result Tensorflow operations only recognize CPU devices. we are forced to downgrade to a previous stable image version as a workaround as GPU support is required, but ideally latest should be used.

Looking at the gcr.io container registry for this image gcr.io/cos-cloud/cos-gpu-installer:latest there has been a new release of a latest and a v2.0.0 version release as of 22/07/2020, neither of these images install graphics drivers. downgrading to v20200701 is installing drivers as expected.

Our compute instances are using 2x Tesla P100-PCIE-16GB FYI

Using latest

Results in the below log on instance start up:

image

Tensor flow only is able recognise CPUs using latest:

image

Using v20200701

We are able to see GPU drivers installed and GPUs are visible using Nvidia-smi

image

Tensorflow is then able recognise GPUs using v20200701:

image

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions