update

Author: sungjin.712

Dockerfile_gpu

@@ -28,9 +28,9 @@ RUN pip install konlpy cmake
 
 # RUN pip install gensim soynlp soyspacing bokeh networkx selenium lxml pyldavis sentencepiece
 
-COPY requirements.txt .
-RUN pip install -r requirements.txt
-RUN rm requirements.txt
+COPY requirements_gpu.txt .
+RUN pip install -r requirements_gpu.txt
+RUN rm requirements_gpu.txt
 RUN pip install setGPU
 # RUN pip install numpy jupyter pandas html5lib seaborn matplotlib nltk tqdm transformers mxnet gluonnlp sklearn wordcloud
 RUN python -c 'import nltk; nltk.download("punkt"); nltk.download("stopwords")'

exec_jupyter_for_gpu.sh

@@ -1,3 +1,3 @@
-sudo docker run -it --rm -p 8889:8889 --gpus=all --memory=4g --memory-swap=8g\
+sudo docker run -it --rm -p 8889:8889 --gpus=all --memory=12g --memory-swap=24g\
  --mount type=bind,destination=/workspace/practice,source=`pwd`\
  tf2_practice_gpu

requirements.txt

@@ -17,7 +17,7 @@ gensim
 xgboost
 wget==3.2
 tokenizers==0.8.1.rc1
-tensorflow==2.0.3
+tensorflow==2.2.0
 h5py<3.0.0              # to avoid issue https://github.com/keras-team/keras/issues/14265
 konlpy
 beautifulsoup4

requirements_gpu.txt

@@ -0,0 +1,24 @@
+numpy<=1.16.2           # to avoid issue https://stackoverflow.com/questions/55890813/how-to-fix-object-arrays-cannot-be-loaded-when-allow-pickle-false-for-imdb-loa
+jupyter 
+pandas 
+html5lib 
+seaborn 
+matplotlib<3.3          # to avoid MatplotlibDeprecationWarning
+nltk==3.3 
+tqdm
+seqeval 
+transformers==3.0.2
+sentencepiece==0.1.85
+mxnet==1.6.0
+gluonnlp==0.9.1
+scikit-learn>=0.19.1    # package name registered in PyPI.org
+wordcloud
+gensim
+xgboost
+wget==3.2
+tokenizers==0.8.1.rc1
+tensorflow-gpu==2.0.3
+h5py<3.0.0              # to avoid issue https://github.com/keras-team/keras/issues/14265
+konlpy
+beautifulsoup4
+kaggle