Ubuntu+GPU上Debug Torch项目fairseq

前端之家收集整理的这篇文章主要介绍了Ubuntu+GPU上Debug Torch项目fairseq前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

1,下载并启动zbstudio

  1. $ git clone https://github.com/soumith/zbs-torch.git
  2. $ cd zbs-torch
  3. $ ./zbstudio.sh

2,安装Torch
http://torch.ch/docs/getting-started.html

3,安装nccl

  1. git clone https://github.com/NVIDIA/nccl
  2. cd nccl
  3. make install
  4. sudo ldconfig
  5. luarocks install nccl

找到libnccl.so*的目录在/etc/bashrc里加上

  1. export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64:/usr/local/lib"

3,安装fairseq
https://github.com/facebookresearch/fairseq#requirements-and-installation

4,把fairseq导入zbstudio
在zbstudio里选择project->project directory->choose

5,preprocess数据
https://github.com/facebookresearch/fairseq#data-pre-processing

6,开始Debug 在train.lua里修改datadir参数指向fairseq/data-bin/iwslt14.tokenized.de-en 运行train.lua,也选中train.lua就是点那个绿色三角

猜你在找的Ubuntu相关文章