aglinxinyuan commented on code in PR #4473: URL: https://github.com/apache/texera/pull/4473#discussion_r3127488983
########## 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: Why do we need a condition check here? Can we just replace torch==2.8.0 with torch==2.8.0+cpu? -- 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]
