Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Install mkl for PyTorch and libjpeg/libpng for TorchVision #8248

Merged
merged 15 commits into from
Feb 10, 2025
Prev Previous commit
Next Next commit
Typo
huydhn committed Feb 7, 2025
commit 726b38d12c7bae81e1302523019392beb44e019c
2 changes: 1 addition & 1 deletion .ci/docker/common/install_base.sh
Original file line number Diff line number Diff line change
@@ -24,7 +24,7 @@ install_ubuntu() {
gdb \
rsync \
libssl-dev \
zip \
zip

# These libraries are needed by TorchVision to get optimal performance processing images
apt-get install -y --no-install-recommends \
2 changes: 1 addition & 1 deletion .ci/docker/common/install_conda.sh
Original file line number Diff line number Diff line change
@@ -38,7 +38,7 @@ install_python() {
if [[ $(uname -m) == "aarch64" ]]; then
conda_install "openblas==0.3.28=*openmp*"
else
conda_install mkl=2021.4.0 mkl-include=2021.4.0
conda_install mkl=2022.1.0 mkl-include=2022.1.0
fi

popd