LeakGANをインストールしサンプルを動かすまで[GAN][NLP]

7月 28, 2020

長文を生成可能なLeakGANをインストールし，サンプルを動かしてみたのでメモ．本記事では，dockerで作成したコンテナ上でサンプルを動かす手順について紹介．

まずはnvidia-docker2で動作環境を用意する．

docker run -it --runtime=nvidia nvidia/cuda:7.5-cudnn5-devel-ubuntu14.04

コンテナに入り，パッケージをインストール．

# apt update
# apt install git
# apt install unzip

Anacondaに必要パッケージのインストール．

# apt install -y make build-essential libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm libncurses5-dev libncursesw5-dev libpng-dev

pyenvのインストール．

# cd ~
# mkdir git
# cd git
# git clone git://github.com/yyuu/pyenv.git ~/.pyenv
# git clone https://github.com/yyuu/pyenv-pip-rehash.git ~/.pyenv/plugins/pyenv-pip-rehash
# echo 'export PYENV_ROOT="$HOME/.pyenv"' >> ~/.bashrc
# echo 'export PATH="$PYENV_ROOT/bin:$PATH"' >> ~/.bashrc
# echo 'eval "$(pyenv init -)"' >> ~/.bashrc
# source ~/.bashrc

Anaconda(Python2系)のバージョンを確認．

# pyenv install -l | grep anaconda2
anaconda2-2.4.0
anaconda2-2.4.1
anaconda2-2.5.0
anaconda2-4.0.0
anaconda2-4.1.0
anaconda2-4.1.1
anaconda2-4.2.0
anaconda2-4.3.0
anaconda2-4.3.1
anaconda2-4.4.0
anaconda2-5.0.0
anaconda2-5.0.1
anaconda2-5.1.0
anaconda2-5.2.0

Anacondaをインストール．

# pyenv install anaconda2-2.4.0
# pyenv global anaconda2-2.4.0
# echo 'export PATH="$PYENV_ROOT/versions/anaconda2-2.4.0/bin:$PATH"' >> ~/.bashrc
# source ~/.bashrc

Tensorflow-gpu 1.2.1 をインストール．

pip install tensorflow-gpu==1.2.1

適当なディレクトリにLeakGANを落としてくる．

git clone https://github.com/CR-Gjx/LeakGAN.git

サンプルを実行．

cd Image COCO
python Main.py

サンプルを実行後，Image COCO/speech にファイルが出力される．

convert.py を使って，出力ファイルをテキストファイルに変換することができる．

GAN,NLP

Posted by vastee