Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Reducing space of the CPU docker image #11807

Merged
merged 25 commits into from
Mar 9, 2023
Merged

Reducing space of the CPU docker image #11807

Changes from 12 commits
Commits
Show all changes
25 commits
Select commit Hold shift + click to select a range
8b9355e
Install CPU only version of torch, as there is no need to install cuD…
ZiadAmerr Mar 6, 2023
4d07fdf
Install CPU only version of torch, as there is no need to install cuD…
ZiadAmerr Mar 7, 2023
36e3f73
Install CPU only version of torch
ZiadAmerr Mar 7, 2023
4319adf
Update Dockerfile
ZiadAmerr Mar 8, 2023
393a7fc
Updated torch installation to install CPU only
ZiadAmerr Mar 8, 2023
dc5cbec
Updated torch installation to install CPU only
ZiadAmerr Mar 8, 2023
59527e6
Install CPU-only torch
ZiadAmerr Mar 8, 2023
ed7eae2
Updated torch installation
ZiadAmerr Mar 8, 2023
cfb54b3
Pinned torch to latest
ZiadAmerr Mar 8, 2023
b6b5278
Unpinned versions of torch CPU
ZiadAmerr Mar 8, 2023
173ba8f
Unpinned versions of torch CPU
ZiadAmerr Mar 8, 2023
5d0aeab
Merge branch 'master' into dockerfileCPU_torch
sagewhocodes Mar 8, 2023
d24bbbb
Updated Dockerfile
ZiadAmerr Mar 8, 2023
a2dc427
Remove nvidia-ml-py library
ZiadAmerr Mar 8, 2023
84233ea
Delete requirements_cpu.txt
ZiadAmerr Mar 8, 2023
47b2491
Added `optional_cpu.txt`
ZiadAmerr Mar 8, 2023
56768f2
Merge remote-tracking branch 'origin/dockerfileCPU_torch' into docker…
ZiadAmerr Mar 8, 2023
d9be29e
Updated `optional_cpu.txt`'s comments
ZiadAmerr Mar 8, 2023
3c9bd01
Delete optional_cpu.txt
ZiadAmerr Mar 8, 2023
bf14350
Install only cpu speific variant libraries
Dsantra92 Mar 8, 2023
2161ed5
Minor change in Dockerfile
ZiadAmerr Mar 8, 2023
d9ce460
Reverted back
ZiadAmerr Mar 8, 2023
f322bbf
Defaulted to CPU-based libraries
ZiadAmerr Mar 9, 2023
8a62a74
test with optional_gpu
Dsantra92 Mar 9, 2023
6bbf566
remove optional_gpu
Dsantra92 Mar 9, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion docker/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,8 @@ RUN apt-get update && \
pip3 install --upgrade pip && \
pip3 install setuptools==58.5.3

RUN pip3 install --no-cache-dir torch -f https://download.pytorch.org/whl/torch_stable.html

RUN pip3 install --no-cache-dir torch --extra-index-url https://download.pytorch.org/whl/cpu
RUN pip3 install --no-cache-dir --upgrade torch-scatter -f https://pytorch-geometric.com/whl

# Install Ivy Upstream
Expand Down