Commit 586a738
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
@@ -43,7 +43,7 @@ RUN conda config --add channels conda-forge && \ | |||
| 43 | 43 | # b/182405233 pyproj 3.x is not compatible with basemap 1.2.1 | |
| 44 | 44 | # b/161473620#comment7 pin required to prevent resolver from picking pysal 1.x., pysal 2.2.x is also downloading data on import. | |
| 45 | 45 | conda install matplotlib basemap cartopy python-igraph imagemagick "pyproj=2.6" "pysal==2.1.0" && \ | |
| 46 | - conda install "pytorch=1.7" "torchvision=0.8" "torchaudio=0.7" "torchtext=0.8" cpuonly && \ | ||
| 46 | + conda install "pytorch==1.8.1" "torchvision==0.9.1" "torchaudio==0.8.1" "torchtext==0.9.1" cpuonly && \ | ||
| 47 | 47 | /tmp/clean-layer.sh | |
| 48 | 48 | ||
| 49 | 49 | # 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 | |
|---|---|---|---|
@@ -59,7 +59,7 @@ RUN apt-get install -y ocl-icd-libopencl1 clinfo libboost-all-dev && \ | |||
| 59 | 59 | # However, because this image is based on the CPU image, this isn't possible but better | |
| 60 | 60 | # to put them at the top of this file to minize conflicts. | |
| 61 | 61 | RUN conda remove --force -y pytorch torchvision torchaudio cpuonly && \ | |
| 62 | - conda install "pytorch=1.7" "torchvision=0.8" "torchaudio=0.7" "torchtext=0.8" "cudf=0.16" "cuml=0.16" cudatoolkit=$CUDA_VERSION && \ | ||
| 62 | + conda install "pytorch==1.8.1" "torchvision==0.9.1" "torchaudio==0.8.1" "torchtext==0.9.1" "cudf=0.16" "cuml=0.16" cudatoolkit=$CUDA_VERSION && \ | ||
| 63 | 63 | /tmp/clean-layer.sh | |
| 64 | 64 | ||
| 65 | 65 | # Install LightGBM with GPU | |
0 commit comments