1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42
|
#!/bin/bash
if [[ "$(uname)" == Darwin ]]; then
# Uninstall Conflicting jpeg brew formulae
jpeg_packages=$(brew list | grep jpeg)
echo "Existing Jpeg-related Brew libraries"
echo $jpeg_packages
for pkg in $jpeg_packages; do
brew uninstall --ignore-dependencies --force $pkg || true
done
conda install -yq wget
fi
if [[ "$(uname)" == Darwin || "$OSTYPE" == "msys" ]]; then
conda install libpng libwebp -yq
# Installing webp also installs a non-turbo jpeg, so we uninstall jpeg stuff
# before re-installing them
conda uninstall libjpeg-turbo libjpeg -y
conda install -yq ffmpeg=4.2 libjpeg-turbo -c pytorch
# Copy binaries to be included in the wheel distribution
if [[ "$OSTYPE" == "msys" ]]; then
python_exec="$(which python)"
bin_path=$(dirname $python_exec)
cp "$bin_path/Library/bin/libjpeg.dll" torchvision
fi
else
if [[ "$ARCH" == "aarch64" ]]; then
conda install libpng -yq
conda install -yq ffmpeg=4.2 libjpeg-turbo -c pytorch-nightly
fi
conda install libwebp -yq
conda install libjpeg-turbo -c pytorch
yum install -y freetype gnutls
pip install auditwheel
fi
pip install numpy pyyaml future ninja
pip install --upgrade setuptools==72.1.0
|