Keine NVIDIA-Pakete im CPU-Build

Als Dev will ich die Python-Pakete von NVIDIA aus dem CPU-Build entfernen, weil

  1. sie für CPU nicht benötigt werden
  2. das Image dadurch kleiner wird
  3. 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
To upload designs, you'll need to enable LFS and have an admin enable hashed storage. More information