Skip to content

Update driver version of EC2 image #1254

@dsuess

Description

@dsuess

Description

The current docker-images use CUDA 10.1, since that seems to be the least common denominator between all the packages used. However, when we tried to run pytorch with CUDA 10.2 (necessary because of pytorch extensions compiled with 10.2), we're getting a runtime error

RuntimeError: cuda runtime error (35) : CUDA driver version is insufficient for CUDA runtime version ...

Motivation

Pytorch 1.5 and 1.6 come by default compiled against CUDA 10.2 so it would be nice if the underlying EC2 image's driver would be compatible with CUDA 10.2.

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions