gaspersic/gpt-2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add a Dockerfile and document usage in README

Browse Source
This commit is contained in:
Madison May
2019-02-14 18:22:14 -05:00
committed by Jeff Wu
parent 23ed990b03
commit 99af6d7092
3 changed files with 50 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
README.md
View File

@ -13,6 +13,8 @@ Git clone this repository, and `cd` into directory for remaining commands
git clone https://github.com/openai/gpt-2.git && cd gpt-2
```
### Native Installation
Download the model data
```
sh download_model.sh 117M
@ -34,6 +36,21 @@ Install other python packages:
pip3 install -r requirements.txt
```
### Docker Installation
Build the Dockerfile and tag the created image as `gpt-2`:
```
docker build --tag gpt-2 -f Dockerfile.gpu . # or Dockerfile.cpu
```
Start an interactive bash session from the `gpt-2` docker image.
You can opt to use the `--runtime=nvidia` flag if you have access to a NVIDIA GPU
and a valid install of [nvidia-docker 2.0](https://github.com/nvidia/nvidia-docker/wiki/Installation-(version-2.0)).
```
docker run --runtime=nvidia -it gpt-2 bash
```
## Usage
| WARNING: Samples are unfiltered and may contain offensive content. |