地平线机器人——嵌入式人工智能领导者 长期招聘计算机视觉/深度学习等方向的工程师或实习生。感兴趣的话,请发送邮件,可以内推。
在Docker上运行MXNet
Docker 是一个系统,它允许你构建一个自包含的Linux操作系统,该操作系统可以在你的计算机上孤立的运行。 在自包含的Linux系统中,你可以运行MXNet和其它软件包,而这些软件不会与你的计算机上的包相互作用。
MXNet 提供了2个运行MXNet的 Docker 镜像:
MXNet Docker (CPU版本) - https://hub.docker.com/r/kaixhin/mxnet/
MXNet Docker (GPU版本) - https://hub.docker.com/r/kaixhin/cuda-mxnet/
每周都会使用MXNet的最新构建,更新这些Docker镜像。
如果需要支持 CUDA,你需要 NVIDIA Docker image。
在Docker上运行MXNet的步骤:
- 安装Docker。更多信息,查看 Docker documentation。
- 运行下面的命令,启动 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。