docker ip共享与gpu指定
1)ip共享
docker网络有多种,这里选择host直接用宿主机的ip
2)指定gpu
docker方式进行pytorch多机多卡分布式训练
nvidia docker 容器介绍链接:https://docs.nvidia.com/deeplearning/frameworks/pytorch-release-notes/
docker用的nvidia官方镜像上面进行继续相关包的安装增加成(cv-worker:v9 整体有点大19G)
docker run -it --gpus '"device=1"' -p 8167:8167 cv-worker:v9 /bin/bash
指定1卡,主机的第二张卡
python -c "import