根据新闻标题，自动分类模型

• 查看版本对应
  https://pytorch.org/get-started/locally/

https://zhuanlan.zhihu.com/p/508595872

github:
https://github.com/649453932/Bert-Chinese-Text-Classification-Pytorch

启动 docker 容器

docker run -it -d --name pytorch --gpus=all -v /usr/local/src:/workspace pytorch/pytorch

运行代码

docker exec -it pytorch bash
#调用GPU训练模型，运行时间大约需要20分钟
python run.py --model bert
执行代码
python pred.py

　查找镜像地址

pytorch/pytorch