Skip to content

Commit 52c194b

Browse files
committed
fix pytorch install and protobuf
1 parent cd8b808 commit 52c194b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Dockerfile.tmpl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ RUN mamba install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION
115115
# b/255757999 openmp (libomp.so) is an dependency of libtorchtext and libtorchaudio but
116116
# the built from source versions don't seem to properly link it in. This forces the dep
117117
# which makes sure that libomp is loaded when these libraries are loaded.
118-
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 torchtext==0.13.0 --extra-index-url https://download.pytorch.org/whl/cu113 && \
118+
mamba install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 torchtext cudatoolkit=11.3 -c pytorch && \
119119
#pip install patchelf && \
120120
#patchelf --add-needed libomp.so /opt/conda/lib/python3.7/site-packages/torchtext/lib/libtorchtext.so && \
121121
#patchelf --add-needed libomp.so /opt/conda/lib/python3.7/site-packages/torchaudio/lib/libtorchaudio.so && \
@@ -390,7 +390,7 @@ RUN pip install google-cloud-automl==1.0.1 \
390390
google-cloud-language==2.* \
391391
google-cloud-videointelligence==2.* \
392392
google-cloud-vision==2.* \
393-
pip install protobuf==3.20.3
393+
protobuf==3.20.3
394394
# install cython & cysignals before pyfasttext
395395
RUN pip install cython \
396396
cysignals \

0 commit comments

Comments
 (0)