Ubuntu OpenNTM配置

weixin_34413065發表於2019-01-05

寫給更換電腦或系統崩潰後的自己。。

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

相關文章