File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -115,7 +115,7 @@ RUN mamba install -c pytorch magma-cuda${CUDA_MAJOR_VERSION}${CUDA_MINOR_VERSION
115
115
# b/255757999 openmp (libomp.so) is an dependency of libtorchtext and libtorchaudio but
116
116
# the built from source versions don't seem to properly link it in. This forces the dep
117
117
# 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 && \
119
119
#pip install patchelf && \
120
120
#patchelf --add-needed libomp.so /opt/conda/lib/python3.7/site-packages/torchtext/lib/libtorchtext.so && \
121
121
#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 \
390
390
google-cloud-language==2.* \
391
391
google-cloud-videointelligence==2.* \
392
392
google-cloud-vision==2.* \
393
- pip install protobuf==3.20.3
393
+ protobuf==3.20.3
394
394
# install cython & cysignals before pyfasttext
395
395
RUN pip install cython \
396
396
cysignals \
You can’t perform that action at this time.
0 commit comments