关于docker,nvidia-docker,cudnn和tensorflow的问题

时间:2019-05-30 07:40:47

标签: gpu docker

我对cudnn和tensorflow-gpu docker映像有一些疑问。这是我在计算机上所做的:

  1. 我安装了docker-CE
  2. 我安装了nvidia-docker
  3. 我从docker hub提取了tensorflow:latest-gpu-py3-jupyter张图片。
  4. 我使用以下命令创建一个容器:

    sudo docker create -ti --rumtime=nvidia -p 21001:22 -v
    /home/project:/project tensorflow:latest-gpu-py3-jupyter /bin/bash
    
  5. 我开始进入容器,然后训练使用Keras构建的模型。但我收到一条错误消息-“分段错误(核心已转储)”

似乎cudnn未安装在我的容器中。有人知道为什么会这样吗?张量流图像不包括cudnn库吗?

0 个答案:

没有答案