写给更换电脑或系统崩溃后的自己。。
首先从github上下载OpenNMT,这里使用的是pytorch版本,注意现在只支持pytorch1.0以上版本了
https://github.com/OpenNMT/OpenNMT-py
如果没有服务器的root权限,先在自己的个人路径中安装python或anaconda后将自己的环境变量设置到你安装的anaconda或python上,不然在自带的python基础上安装环境需求是需要root权限的
下载解压后,进入目录,配置requirements
pip install -r requirements.txt
这一步安装text会失败,应该是github源失效的缘故,更改requirements.txt的内容,删除带有text的那一行重新安装,之后单独安装text即可。单独安装输入:
pip install torchtext
之后就可以安装官网上的教程进行训练了~
如果想使用gpu,在训练的命令后加上
-gpu_rank gpuid
如果要使用多个,在指令python前加上字段CUDA_VISIBLE_DEVICES=id1,id2...
gpuid是你要使用的gpu的编号,相对与前面加的id1、id2从0开始
可以使用一下命令查看各个gpu的编号和使用情况,按需选择
nvidia-smi