diff --git a/docker/caffe2/jenkins/common/install_rocm.sh b/docker/caffe2/jenkins/common/install_rocm.sh index e93a5bc9557b07..2910390a6af4df 100644 --- a/docker/caffe2/jenkins/common/install_rocm.sh +++ b/docker/caffe2/jenkins/common/install_rocm.sh @@ -5,6 +5,7 @@ set -ex install_ubuntu() { apt-get update apt-get install -y wget + apt-get install -y libopenblas-dev DEB_ROCM_REPO=http://repo.radeon.com/rocm/apt/debian # Add rocm repository