Skip to content
This repository was archived by the owner on Aug 15, 2025. It is now read-only.

Conversation

seemethere
Copy link
Member

Adds an option to build conda builder images with only one version of
CUDA installed effectively reducing image sizes from 20GB+ to ~10GB

Signed-off-by: Eli Uriegas eliuriegas@fb.com

ADD ./java/jni.h /usr/local/include/jni.h
ENV PATH /opt/conda/bin:$PATH
COPY --from=mnist /usr/local/mnist /usr/local/mnist
RUN rm -rf /usr/local/cuda
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why did we need to remove /usr/local/cuda before and do we need to do that after line 73 too?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

it's to remove the original /usr/local/cuda from the base nvidia image

@seemethere seemethere force-pushed the singular_conda_images branch from 597727b to c4987f2 Compare February 1, 2021 21:01
Adds an option to build conda builder images with only one version of
CUDA installed effectively reducing image sizes from 20GB+ to ~10GB

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>
@seemethere seemethere force-pushed the singular_conda_images branch from c4987f2 to f9592c8 Compare February 1, 2021 21:02
@seemethere seemethere merged commit 0b545ab into pytorch:master Feb 1, 2021
@seemethere seemethere deleted the singular_conda_images branch February 1, 2021 21:03
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 1, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 2, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 3, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 3, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 3, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 4, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 4, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 4, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 4, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 4, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 4, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 4, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 5, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 5, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Differential Revision: [D26281958](https://our.internmc.facebook.com/intern/diff/D26281958)

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 10, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Differential Revision: [D26281958](https://our.internmc.facebook.com/intern/diff/D26281958)

[ghstack-poisoned]
seemethere added a commit to pytorch/pytorch that referenced this pull request Feb 26, 2021
The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Differential Revision: [D26281958](https://our.internmc.facebook.com/intern/diff/D26281958)

[ghstack-poisoned]
facebook-github-bot pushed a commit to pytorch/pytorch that referenced this pull request Feb 26, 2021
Summary:
Pull Request resolved: #51494

The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Test Plan: Imported from OSS

Reviewed By: janeyx99

Differential Revision: D26281958

Pulled By: seemethere

fbshipit-source-id: 83b498532a6f04801952438537b564f998b62d94
aocsa pushed a commit to Quansight/pytorch that referenced this pull request Mar 15, 2021
Summary:
Pull Request resolved: pytorch#51494

The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Test Plan: Imported from OSS

Reviewed By: janeyx99

Differential Revision: D26281958

Pulled By: seemethere

fbshipit-source-id: 83b498532a6f04801952438537b564f998b62d94
xsacha pushed a commit to xsacha/pytorch that referenced this pull request Mar 31, 2021
Summary:
Pull Request resolved: pytorch#51494

The overall `pytorch/conda-cuda` image was getting to a ridiculous size
of 36GB so this splits up that image into cuda specific ones to try and
reduce the amount of things we have to download.

coincides with: pytorch/builder#634

Signed-off-by: Eli Uriegas <eliuriegas@fb.com>

Test Plan: Imported from OSS

Reviewed By: janeyx99

Differential Revision: D26281958

Pulled By: seemethere

fbshipit-source-id: 83b498532a6f04801952438537b564f998b62d94
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants