-
Notifications
You must be signed in to change notification settings - Fork 607
Closed
Labels
enhancementNew feature or requestNew feature or request
Description
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
Labels
enhancementNew feature or requestNew feature or request