Ubuntu OpenNTM配置

写给更换电脑或系统崩溃后的自己。。

首先从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

你可能感兴趣的:(Ubuntu OpenNTM配置)