maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

pynvml added to Dockerfile

Browse source
This commit is contained in:
Abhishek Sawarkar 2021-03-03 13:59:13 -08:00
parent b181af5e5b
commit edb16c5e12
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
TensorFlow/LanguageModeling/BERT/Dockerfile
View file

@ -4,7 +4,7 @@ FROM ${FROM_IMAGE_NAME}
RUN apt-get update && apt-get install -y pbzip2 pv bzip2 libcurl4 curl libb64-dev
RUN pip install --upgrade pip
RUN pip install toposort networkx pytest nltk tqdm html2text progressbar
RUN pip install toposort networkx pytest nltk tqdm html2text progressbar pynvml
RUN pip --no-cache-dir --no-cache install git+https://github.com/NVIDIA/dllogger wget
WORKDIR /workspace