File tree Expand file tree Collapse file tree 3 files changed +10
-6
lines changed Expand file tree Collapse file tree 3 files changed +10
-6
lines changed Original file line number Diff line number Diff line change @@ -106,7 +106,7 @@ RUN conda install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION
106
106
rm -rf /tmp/torch && \
107
107
/tmp/clean-layer.sh
108
108
{{ else }}
109
- RUN pip install torch==$TORCH_VERSION+cpu torchvision==$TORCHVISION_VERSION+cpu torchaudio==$TORCHAUDIO_VERSION torchtext==$TORCHTEXT_VERSION -f https://download.pytorch.org/whl/torch_stable.html && \
109
+ RUN pip install torch==$TORCH_VERSION+cpu torchvision==$TORCHVISION_VERSION+cpu torchaudio==$TORCHAUDIO_VERSION+cpu torchtext==$TORCHTEXT_VERSION -f https://download.pytorch.org/whl/torch_stable.html && \
110
110
/tmp/clean-layer.sh
111
111
{{ end }}
112
112
@@ -497,7 +497,8 @@ RUN pip install flashtext && \
497
497
pip install bqplot && \
498
498
pip install earthengine-api && \
499
499
pip install transformers && \
500
- pip install datasets && \
500
+ # b/232247930 >= 2.2.0 requires pyarrow >= 6.0.0 which conflicts with dependencies for rapidsai 0.21.*
501
+ pip install datasets==2.1.0 && \
501
502
pip install dlib && \
502
503
pip install kaggle-environments && \
503
504
pip install geopandas && \
Original file line number Diff line number Diff line change @@ -3,9 +3,10 @@ BASE_IMAGE_TAG=m91
3
3
CPU_BASE_IMAGE_NAME=tf2-cpu.2-6
4
4
GPU_BASE_IMAGE_NAME=tf2-gpu.2-6
5
5
LIGHTGBM_VERSION=3.3.1
6
- TORCH_VERSION=1.9.1
7
- TORCHAUDIO_VERSION=0.9.1
8
- TORCHTEXT_VERSION=0.10.1
9
- TORCHVISION_VERSION=0.10.1
6
+ TORCH_VERSION=1.11.0
7
+ # TODO(b/215031404#comment4) Remove zlib sed command after upgrade to >= 0.11.1
8
+ TORCHAUDIO_VERSION=0.11.0
9
+ TORCHTEXT_VERSION=0.12.0
10
+ TORCHVISION_VERSION=0.12.0
10
11
CUDA_MAJOR_VERSION=11
11
12
CUDA_MINOR_VERSION=0
Original file line number Diff line number Diff line change @@ -44,6 +44,8 @@ RUN cd /usr/local/src && \
44
44
git checkout tags/v$TORCHAUDIO_VERSION && \
45
45
git submodule sync && \
46
46
git submodule update --init --recursive --jobs 0 && \
47
+ # TODO(b/215031404#comment4) Remove after upgrade next release (0.11.1)
48
+ sed -i s?https://zlib.net/zlib-1.2.11.tar.gz?https://sourceforge.net/projects/libpng/files/zlib/1.2.11/zlib-1.2.11.tar.gz? third_party/zlib/CMakeLists.txt && \
47
49
python setup.py bdist_wheel
48
50
49
51
# Build torchtext
You can’t perform that action at this time.
0 commit comments