diff --git a/scripts/build_torch_wheels.sh b/scripts/build_torch_wheels.sh index 2ac573f7faa8..a392097485a6 100755 --- a/scripts/build_torch_wheels.sh +++ b/scripts/build_torch_wheels.sh @@ -129,7 +129,7 @@ function install_llvm_clang() { } function install_req_packages() { - sudo apt-get -y install python-pip git curl libopenblas-dev vim apt-transport-https ca-certificates wget + sudo apt-get -y install python-pip git curl libopenblas-dev vim apt-transport-https ca-certificates wget procps maybe_install_cuda install_bazel }