欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-04-04 21:20:25
Ubuntu14.04系统下,Torch7的安装教程 本文基于Ubuntu14.04,linux系统安装不赘述。 需要使用GPU加速,需要安装cuda以及cudnn模块。cud...

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.即可进入网页的代码编辑界面。