MxNet系列——get_started——docker_setup

博客新址: http://blog.xuezhisd.top
邮箱:[email protected]


地平线机器人——嵌入式人工智能领导者 长期招聘计算机视觉/深度学习等方向的工程师或实习生。感兴趣的话,请发送邮件,可以内推


在Docker上运行MXNet

Docker 是一个系统,它允许你构建一个自包含的Linux操作系统,该操作系统可以在你的计算机上孤立的运行。 在自包含的Linux系统中,你可以运行MXNet和其它软件包,而这些软件不会与你的计算机上的包相互作用。

MXNet 提供了2个运行MXNet的 Docker 镜像:

  1. MXNet Docker (CPU版本) - https://hub.docker.com/r/kaixhin/mxnet/

  2. MXNet Docker (GPU版本) - https://hub.docker.com/r/kaixhin/cuda-mxnet/

每周都会使用MXNet的最新构建,更新这些Docker镜像。

如果需要支持 CUDA,你需要 NVIDIA Docker image

在Docker上运行MXNet的步骤:

  1. 安装Docker。更多信息,查看 Docker documentation
  2. 运行下面的命令,启动 MXNet Docker容器。

对于 CPU容器,执行下面的命令:

    sudo docker run -it kaixhin/mxnet

对于 GPU容器,执行下面的命令:

    sudo nvidia-docker run -it kaixhin/cuda-mxnet:7.0

关于如何使用 MXNet Docker镜像的更多细节,请查看 MXNet Docker GitHub documentation

下一步

猜你喜欢

转载自blog.csdn.net/xuezhisdc/article/details/54945319