核心提示:Ubuntu14.04系统下,Torch7的安装教程本文基于Ubuntu14.04,linux系统安装不赘述。需要使用GPU加速,需要安装cuda以及cudnn模块。cuda安装完毕后才可以安装cut...
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.即可进入网页的代码编辑界面。