Keine NVIDIA-Pakete im CPU-Build
Als Dev will ich die Python-Pakete von NVIDIA aus dem CPU-Build entfernen, weil
- sie für CPU nicht benötigt werden
- das Image dadurch kleiner wird
- die Build-Zeiten kürzer werden
Vorschlag Vorgehen
Wir verwenden die aktuelle requirements.{in,txt} nur für GPU-Builds (welche auf Opencode nicht stattfinden). Damit die Verwendung explizit wird, nennen wir diese requirements-gpu.{in,txt}.
Daneben haben wir eine neue requirements.{in,txt}, die mit dem --extra-index-url=https://download.pytorch.org/whl/cpu gebaut werden. Diese enthält keine NVIDIA-Pakete mehr.
requirements.in --> requirements-gpu.in
requirements.txt --> requirements-gpu.txt
# CPU-Variante der Requirements
requirements.in mit --extra-index-url=https://download.pytorch.org/whl/cpu
pip-compile requirements.in
Akzeptanzkriterien
-
requirements-gpu.{in,txt} angelegt/gebaut (pip-compile) -
docker-compose-gpu.yml: Build Section ergänzen mit REQUIREMENTS_FILE=requirements-gpu.txt -
torch-Version in der requirements.in stimmt mit der Im Base-Image überein (um Neuinstallation zu verhindern) -
Anpassung der Dokumentation für den Bau eines GPU Image -
Changelog angepasst -
Tests laufen durch -
Review einer zweiten Person durchgeführt
Edited by Denny Mattern