dockerpytorch的简单介绍

本篇文章给大家谈谈dockerpytorch，以及对应的知识点，希望对各位有所帮助，不要忘了收藏本站喔。

本文目录一览：

• 1、docker容器内运行pytorch多gpu报错 RuntimeError: NCCL Error 2: unhandled system error
• 2、docker完全卸载与安装
• 3、Docker - nvidia使用

docker容器内运行pytorch多gpu报错 RuntimeError: NCCL Error 2: unhandled system error

错误1.docker容器内运行pytorch多gpu报错 RuntimeError: NCCL Error 2: unhandled system error

在启动容困郑器的时候加上 -e NVIDIA_VISIBLE_DEVICES=0,1,2,3

docker run --runtime=nvidia --net="host" -e NVIDIA_VISIBLE_DEVICES=0,1,2,3 --shm-size 8g -it huangzc/reid:v1 /bin/bash

错误2.RuntimeError: DataLoader worker (pid 53617) is killed by signal: Bus error. It is possible that dataloader'汪老颂s workers are out of shared memory. Please try to raise your shared memory limit.

启动容器的时候含枣增加交换内存 --shm-size 8g

docker完全卸载与安装

费了九牛二虎之力才卸载成功扰碧

这个帖缓知举子总结了彻底卸载方法，猛中但对我不好用 这里

因为我的docker使用snap安装的，所以需要snap卸载：

sudo snap list

sudo snap remove snap name

sudo snap remove --purge docker

sudo apt install docker.io

拉pytorch总中断，参考 这里 后面的步骤

[img]

Docker - nvidia使用

nvidia cuda使模厅衡用要安装nvidia-docker， 官网 安装介绍非常详细,下面是ubuntu下的安装：旦做

安伏乎装后测试

如果返回正常的nvidia显卡界面则安装成功。

下面我们使用需要的镜像来进行定制。由于要使用pytorch和gpu，我们去dockerhub搜索镜像，最终选择：nvidia/cuda:10.2-cudnn7-devel-ubuntu18.04.

然后就可以根据需要安装必备库了！

关于dockerpytorch和的介绍到此就结束了，不知道你从中找到你需要的信息了吗 ？如果你还想了解更多这方面的信息，记得收藏关注本站。