如何解决Pytorch + Docker:多名工人
我有一个使用pytorch进行远程培训的机器学习项目。但是,docker容器未使用标志(--ipc = host或--shm-size)启动。 pytorch文档(https://github.com/pytorch/pytorch#docker-image)表示,这是在Docker容器中运行多个工作程序所必需的。将工作人员的数量设置为0并不是一个适当的解决方案,因为培训花费的时间要长10倍。有什么方法可以让pytorch数据加载器在Docker容器中工作?我们可以通过自己定义Dockerfile。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。