aglinxinyuan commented on code in PR #4473: URL: https://github.com/apache/texera/pull/4473#discussion_r3127494721
########## amber/operator-requirements.txt: ########## @@ -20,8 +20,12 @@ plotly==5.24.1 praw==7.6.1 pillow==12.1.1 pybase64==1.3.2 -torch==2.8.0 + +# Pin torch to the CPU wheel on Linux x86_64 to avoid the NVIDIA CUDA deps. +--extra-index-url https://download.pytorch.org/whl/cpu +torch==2.8.0+cpu ; platform_system == "Linux" and platform_machine == "x86_64" +torch==2.8.0 ; platform_system != "Linux" or platform_machine != "x86_64" Review Comment: The PR description mentioned that "Other platforms fall through to plain torch==2.8.0 from PyPI, which is already a CPU build." Can we still use torch==2.8.0+cpu across all platforms to simplify this config? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
