当前位置: 移动技术网 > IT编程>网页制作>CSS > Ubuntu14.04系统下,Torch7的安装教程

Ubuntu14.04系统下,Torch7的安装教程

2018年02月15日  | 移动技术网IT编程  | 我要评论

Ubuntu14.04系统下,Torch7的安装教程

本文基于Ubuntu14.04,linux系统安装不赘述。

需要使用GPU加速,需要安装cuda以及cudnn模块。cuda安装完毕后才可以安装cutorch以及cunn、cudnn库。

CUDA下载地址:https://developer.nvidia.com/cuda-downloads

Cudnn下载地址(需注册):https://developer.nvidia.com/cudnn

CUDA安装:下载的为deb离线包

sudo dpkg -i cuda-repo-ubuntu1404-7-5-local_7.5-18_amd64.deb

sudo apt-get update

sudo apt-get install cuda

Cudnn安装:

tar -xzvf ccudnn-7.5-linux-x64-v5.1.tgz

cd cudnn-7.5-linux-x64-v5.1

sudo cp lib* /usr/local/cuda/lib64/

sudo cp cudnn.h /usr/local/cuda/include/

然后就可以安装cutorch、cunn、cudnn库:

luarocks install cutorch

luarocks install cunn

luarocks install cudnn

一般的代码编辑器不能自动补全lua代码,推荐使用itorch notebook进行编辑

安装完毕后终端输入:itorch notebook.即可进入网页的代码编辑界面。

如对本文有疑问, 点击进行留言回复!!

相关文章:

验证码:
移动技术网