您的位置:首页 > 其它

ubuntu14.04 Torch7安装与配置GPU

2017-05-26 22:58 627 查看
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.10
sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -sf libcudnn.so.5 libcudnn.so
sudo ldconfig


2、安装Torch

git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/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即为安装成功
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  ubuntu cuda gpu torch