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