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

最后编辑于 :
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容