We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 74094d6 commit 1563489Copy full SHA for 1563489
1 file changed
ML-Frameworks/pytorch-aarch64/build.sh
@@ -50,7 +50,7 @@ torch_wheel_name=$(grep -o "torch-.*.whl" $build_log | head -n -1 | tail -n 1)
50
# Use the second to last match, otherwise grep finds itself
51
torch_ao_wheel_name=$(grep -o "torchao-.*.whl" $build_log | head -n -1 | tail -n 1)
52
53
-DOCKER_BUILDKIT=1 docker build -t toolsolutions-pytorch:latest \
+docker build -t toolsolutions-pytorch:latest \
54
--build-arg TORCH_WHEEL=results/$torch_wheel_name \
55
--build-arg DOCKER_IMAGE_MIRROR \
56
--build-arg TORCH_AO_WHEEL=ao/dist/$torch_ao_wheel_name \
0 commit comments