Skip to content

Commit be93291

Browse files
committed
Add USE_XPU=0 flag to pytorch installation in lib.sh
Signed-off-by: Eric Curtin <eric.curtin@docker.com>
1 parent 4a7b1c6 commit be93291

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

scripts/lib.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ get_repo_root() {
5252
}
5353

5454
uv_pip_install_pytorch() {
55-
USE_CUDA=0 USE_CUDNN=0 USE_ROCM=0 USE_MPS=1 uv pip install "$pytorch_dir"
55+
USE_CUDA=0 USE_CUDNN=0 USE_ROCM=0 USE_XPU=0 USE_MPS=1 uv pip install "$pytorch_dir"
5656
}
5757

5858
# Install PyTorch from submodule

0 commit comments

Comments
 (0)