ubuntu14.04 Torch7安装与配置GPU

来源:互联网 发布:mysql运行原理 编辑:程序博客网 时间:2024/06/05 13:08

Torch官方网址:http://torch.ch/
前提已经安装好cuda驱动,可以参考我的博客:ubuntu14.04+cuda7.5安装 官方步骤版

1、安装cudnn

官方地址:https://developer.nvidia.com/cudnn
需要注册和审核,然后下载V5.1版本的cudnn(对应自己的cuda版本下载即可)
下载好的文件夹cuda有 include、lib64两个文件夹
将include中的文件移动/usr/local/cuda/include/
将lib64中的文件移动到/usr/local/cuda/lib64/

#在include文件夹下sudo cp cudnn.h /usr/local/cuda/include/#在lib64文件夹下sudo cp libcudnn* /usr/local/cuda/lib64/

命令进入/usr/local/cuda/lib64/文件夹,建立软连接(当前版本为5.1.10)
建立软链接:

sudo chmod +r libcudnn.so.5.1.10sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5sudo ln -sf libcudnn.so.5 libcudnn.sosudo ldconfig

2、安装Torch

git clone https://github.com/torch/distro.git ~/torch --recursivecd ~/torch bash install-deps./install.sh

第三步install-deps可见安装了OpenBLAS、build-essential、gcc、g++、curl、cmake、libreadline-dev、Git-core、libqt4-core、libqt4-gui、libqt4-dev、libjpeg-dev、libpng-dev、ncurses-dev、imagemagick、libzmq3-dev、gfortran、unzip、gnuplot、gnuplot-x11、ipython依赖包。
第四步需要会自动显示选项添加torch环境变量到.bashrc,选择yes添加torch环境变量到个人用户。

3、扩展包安装

Torch支持丰富扩展包,在/torch/extra/路径下可以查看,如果想要添加其他扩展包,只需执行命令

#安装扩展包$ luarocks install 包名称#查看当前安装包信息$ Luarock list 

4、测试

Torch7 依赖于 LuaJIT 解释器, 启动Torch的最方便的方式是启动一个解释器,输入:

luajit

此方式为非交互式方式,不太方便,在安装torch时已经安装了交互式方式(TREPL),启动方式:

th

可以在当前目录中建立名为“927.lua”的lua脚本文件,内容为
print(“hello”)
使用命令 :

th 927.lua

在终端显示hello即为安装成功

原创粉丝点击