Skip to content

Commit ef5241e

Browse files
sharathtsnv-kkudrynski
authored andcommitted
[BERT/PyT] fix for issue 1249
1 parent e6244d7 commit ef5241e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

PyTorch/LanguageModeling/BERT/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ WORKDIR /workspace
2020

2121
WORKDIR /workspace/bert
2222
RUN pip install --no-cache-dir \
23-
tqdm boto3 requests six ipdb h5py nltk progressbar onnxruntime tokenizers>=0.7\
23+
tqdm boto3 requests six ipdb h5py nltk progressbar onnxruntime==1.12.0 tokenizers>=0.7\
2424
git+https://github.com/NVIDIA/dllogger wget
2525

2626
RUN apt-get install -y iputils-ping

0 commit comments

Comments
 (0)