安装Torch7很简单,只需要几条命令即可,本教程在安装完torch7之后还会download下demo,并运行其中的train-a-digit-classifier。
具体安装步骤如下:
- sudo apt-get install git
- git clone https://github.com/torch/distro.git ~/torch --recursive
- cd torch
- ./install-deps
- ./install.sh
- source ~/.bashrc
- source ~/.profile
- th
执行完以上的步骤应该已经将t7环境搭建完成了,在第八步在终端输入th,即验证t7框架是否安装成功,如果执行最后一步出现如下图所示,则证明安装成功。
接下来我们开始运行demo
- git clone https://github.com/torch/demos.git
- cd demos/train-a-digit-classifier
- luac dataset-mnist.lua
- lua train-on-mnist.lua
执行完以上命令demo就已经运行起来了,但是如果我们不做任何修改就直接运行该代码,那么我们可能会等待很长时间,因为这个程序有一个死循环,应该是运行几百轮之后自动结束,但是我坚持到一百就受不了了,所以对程序进行了一些修改,使用vim打开train-on-mnist.lua,在文件最后我们会看到程序中的死循环,我们将以下代码加入到while循环中再运行程序即可。如下:
if epoch == 30 then
break;
end