Skip to content

Dockerfile CPU Smoke test fails to build torch - #74

Merged
Jacob-Chmura merged 5 commits into
mainfrom
bug/dockerfile_build
Jul 2, 2025
Merged

Dockerfile CPU Smoke test fails to build torch#74
Jacob-Chmura merged 5 commits into
mainfrom
bug/dockerfile_build

Conversation

@Jacob-Chmura

@Jacob-Chmura Jacob-Chmura commented Jun 27, 2025

Copy link
Copy Markdown
Member

Purpose

When changing moduel scope in our root package (#71) the eager import forced us to import torch directly in the Dockefile smoke test. This uncovered the following error:

Screenshot from 2025-06-27 06-58-46

Solution

The error was related to the fact that everything was installed in uv virtual environment instead of pip. To fix this, I went with full pip install so that CI passes. Also, that means users of the image can just work with python and avoid uv (in case they don't want it or have it).

Note: Also found an issue with the GPU dockerfile which has been resolved.

Relevant Prs

Fix #73

Out of scope

  • It would be good to have CPU dockerfile only depend on torch[CPU] instead of installing all the nvidia cuda wheels

@Jacob-Chmura Jacob-Chmura self-assigned this Jun 27, 2025
@Jacob-Chmura
Jacob-Chmura changed the base branch from main to dev/module_scope June 27, 2025 11:01
@Jacob-Chmura Jacob-Chmura changed the title Dockerfile CPU Smoke test fails to buidl torch Dockerfile CPU Smoke test fails to build torch Jun 27, 2025
@Jacob-Chmura
Jacob-Chmura marked this pull request as ready for review June 27, 2025 11:47
@Jacob-Chmura
Jacob-Chmura changed the base branch from dev/module_scope to main July 1, 2025 10:36
@Jacob-Chmura
Jacob-Chmura merged commit 9765c33 into main Jul 2, 2025
5 checks passed
@Jacob-Chmura
Jacob-Chmura deleted the bug/dockerfile_build branch July 2, 2025 16:10
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Dockerfile silently fails to biuld torch on cpu

2 participants