nvidia- docker "nvidia / cuda: 10.1-base-ubuntu18.04" изображение принимает cuda 10.2 вместо 10.1 - PullRequest
0 голосов
/ 17 февраля 2020

У меня есть версия tenorflow 2.1, и я пытаюсь создать образ nvidia docker для работы в некоторых проектах глубокого обучения. Так как тензор потока 2.1 не работает с cuda 10.2, я взял это изображение как базовое изображение: «nvidia / cuda: 10.1-base-ubuntu18.04» для использования cuda 10.1, но он устанавливает cuda 10.2, как вы можете видеть в эта фигура. enter image description here

Кто-нибудь знает, как решить эту проблему? На этом рисунке вы можете увидеть мой докер-файл. enter image description here

...