Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

DockerFile with nccl #56

Closed
mingruimingrui opened this issue Oct 29, 2018 · 4 comments
Closed

DockerFile with nccl #56

mingruimingrui opened this issue Oct 29, 2018 · 4 comments
Labels
enhancement New feature or request question Further information is requested

Comments

@mingruimingrui
Copy link

Hi all, I was wondering if you guys have a sample DockerFile with CUDA, CUDNN and NCCL?
Would be really helpful for pushing things to production! Thanks!

@fmassa
Copy link
Contributor

fmassa commented Oct 29, 2018

We don't have a docker image, but maybe you could take the official PyTorch docker image from https://github.com/pytorch/pytorch#docker-image and adapt it to install maskrcnn-benchmark?

@fmassa fmassa added enhancement New feature or request question Further information is requested labels Oct 29, 2018
@mingruimingrui
Copy link
Author

@fmassa Thanks! I actually found what I was looking for. Apparently nccl and even cudnn can be installed using apt get! https://github.com/uber/horovod/blob/master/Dockerfile

@mingruimingrui
Copy link
Author

I'll be closing this issue now.

@fmassa
Copy link
Contributor

fmassa commented Nov 3, 2018

Also note that there is a open PR which adds a Dockerfile for mask RCNN benchmark, see #78

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
enhancement New feature or request question Further information is requested
Projects
None yet
Development

No branches or pull requests

2 participants