diff options
author | bsilvereagle <bsilvereagle@users.noreply.github.com> | 2023-04-02 15:13:03 -0700 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-03 00:13:03 +0200 |
commit | a0c05164168297c04737936ad0cad849a512547a (patch) | |
tree | 30f6e731817e123d4402f95091aa3e6d320345c8 /tests | |
parent | d8d4e865cd481b18f10508ffee35db903767ef5c (diff) |
Remove torch GPU dependencies from the Docker.full image (#665)
By using `pip install torch --index-url https://download.pytorch.org/whl/cpu`
instead of `pip install torch` we can specify we want to install a CPU-only version
of PyTorch without any GPU dependencies. This reduces the size of the Docker image
from 7.32 GB to 1.62 GB
Diffstat (limited to 'tests')
0 files changed, 0 insertions, 0 deletions