Skip to content

Commit a0b81ee

Browse files
committed
Remove pixman pin
1 parent 0682232 commit a0b81ee

File tree

2 files changed

+8
-5
lines changed

2 files changed

+8
-5
lines changed

Dockerfile

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,6 @@ ENV PROJ_LIB=/opt/conda/share/proj
3939
RUN conda install -c conda-forge matplotlib basemap cartopy python-igraph imagemagick pysal && \
4040
# b/142337634#comment22 pin required to avoid torchaudio downgrade.
4141
conda install -c pytorch pytorch torchvision "torchaudio>=0.4.0" cpuonly && \
42-
# pixman 0.38.0 has a known issue causing issues with packages such as pycairo & openslide.
43-
# See: https://gitlab.freedesktop.org/pixman/pixman/commit/8256c235d9b3854d039242356905eca854a890ba
44-
conda install -c conda-forge --no-deps pixman==0.34 && \
4542
/tmp/clean-layer.sh
4643

4744
# The anaconda base image includes outdated versions of these packages. Update them to include the latest version.

gpu.Dockerfile

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,14 @@ RUN apt-get install -y ocl-icd-libopencl1 clinfo libboost-all-dev && \
5656
echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd && \
5757
/tmp/clean-layer.sh
5858

59+
# When using pip in a conda environment, conda commands should be ran first and then
60+
# the remaining pip commands: https://www.anaconda.com/using-pip-in-a-conda-environment/
61+
# However, because this image is based on the CPU image, this isn't possible but better
62+
# to put them at the top of this file to minize conflicts.
63+
RUN conda remove --force -y pytorch torchvision torchaudio cpuonly && \
64+
conda install -y pytorch torchvision torchaudio cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION -c pytorch && \ && \
65+
/tmp/clean-layer.sh
66+
5967
# Install LightGBM with GPU
6068
RUN pip uninstall -y lightgbm && \
6169
cd /usr/local/src && \
@@ -88,8 +96,6 @@ COPY --from=tensorflow_whl /tmp/tensorflow_gpu/*.whl /tmp/tensorflow_gpu/
8896
RUN pip uninstall -y tensorflow && \
8997
pip install /tmp/tensorflow_gpu/tensorflow*.whl && \
9098
rm -rf /tmp/tensorflow_gpu && \
91-
conda remove --force -y pytorch torchvision torchaudio cpuonly && \
92-
conda install -y pytorch torchvision torchaudio cudatoolkit=$CUDA_MAJOR_VERSION.$CUDA_MINOR_VERSION -c pytorch && \
9399
pip uninstall -y mxnet && \
94100
# b/126259508 --no-deps prevents numpy from being downgraded.
95101
pip install --no-deps mxnet-cu$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION && \

0 commit comments

Comments
 (0)