项目地址：https://github.com/openai/gpt-2
概述：不建议使用docker镜像方式安装，因为过程中存在几个重要关卡不好解决：1、tensorflow1版本升级到tensorflow2；2、mac安装tensorflow方式与window不一致
1、克隆项目到本地
1
| git clone https://github.com/openai/gpt-2.git
|
2、创建工作空间并激活
1
2
| conda create -n gpt2 python=3.10
conda activate gpt2
|
3、安装tensorflow
此过程按照安装tensorflow,按照第2步和第3步进行操作
4、安装依赖
1
2
3
| cd gpt-2
python -m pip install -r requirements.txt
python -m pip install numpy
|
5、下载模型卡[可以下载其中一个，也可以全部下载，使用过程中只会使用一个]
1
2
3
4
| python download_model.py 124M
python download_model.py 355M
python download_model.py 774M
python download_model.py 1558M
|
6、将代码从tensorflow1版本转换为tensorflow2版本
1
2
3
4
| # a、代码转换[tf_upgrade_v2指令为安装tensorflow后自带的指令，专门用于代码写法的转换]
tf_upgrade_v2 --intree src/ --outtree src2/
# b、将src2/下的model.py文件中的4个".value"字符串替换为""
# c、将https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/training/python/training/hparam.py文件下载到src2目录下，并修改model.py文件中的"from tensorflow.contrib.training import HParams"为"from hparam import HParams"
|
7、启动gpt2的模型会话[–model_name 缺省参数为 124M]
1
| python src2/interactive_conditional_samples.py --model_name 345M
|