File tree Expand file tree Collapse file tree 2 files changed +8
-5
lines changed Expand file tree Collapse file tree 2 files changed +8
-5
lines changed Original file line number Diff line number Diff line change @@ -39,9 +39,6 @@ ENV PROJ_LIB=/opt/conda/share/proj
39
39
RUN conda install -c conda-forge matplotlib basemap cartopy python-igraph imagemagick pysal && \
40
40
# b/142337634#comment22 pin required to avoid torchaudio downgrade.
41
41
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 && \
45
42
/tmp/clean-layer.sh
46
43
47
44
# The anaconda base image includes outdated versions of these packages. Update them to include the latest version.
Original file line number Diff line number Diff line change @@ -56,6 +56,14 @@ RUN apt-get install -y ocl-icd-libopencl1 clinfo libboost-all-dev && \
56
56
echo "libnvidia-opencl.so.1" > /etc/OpenCL/vendors/nvidia.icd && \
57
57
/tmp/clean-layer.sh
58
58
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
+
59
67
# Install LightGBM with GPU
60
68
RUN pip uninstall -y lightgbm && \
61
69
cd /usr/local/src && \
@@ -88,8 +96,6 @@ COPY --from=tensorflow_whl /tmp/tensorflow_gpu/*.whl /tmp/tensorflow_gpu/
88
96
RUN pip uninstall -y tensorflow && \
89
97
pip install /tmp/tensorflow_gpu/tensorflow*.whl && \
90
98
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 && \
93
99
pip uninstall -y mxnet && \
94
100
# b/126259508 --no-deps prevents numpy from being downgraded.
95
101
pip install --no-deps mxnet-cu$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION && \
You can’t perform that action at this time.
0 commit comments