参考博客 :
1. 安装 torch 7
git clone https://github.com/torch/distro.git ~/torch --recursive
cd ~/torch
bash install-deps
./install.sh
source ~/.bashrc
th
- 顺利的话就会看到下面的torch界面 .
- 如果中间出现问题,可以多次执行
bash install-deps
和./install.sh
的方法有可能得到解决 . 参考博客中给出了一些具体问题的解决方案,我没有遇到在此就不赘述 .
2. 检查 cuda 和 cudnn
- 查看cuda和cudnn的版本信息,确保配置成功 .
cuda 版本
cat /usr/local/cuda/version.txt
cudnn 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
- 配置成功会看到类似下图的版本信息 .
3. 安装 torch 的 cuda 支持
luarocks install cutorch
luarocks install cunn
luarocks install cudnn
cd ~/torch
./test.sh
- 顺利的话就会看到全PASS通过如下图所示,可能会出现一两个错误,重跑可能就消失了了,根据使用经验来看无伤大雅,可以忽略 .
4. 解决 ‘libcudnn not found in library path’
即使上述
./test.sh
出现 0 failures and 0 errors,最后也会出现这样的提示,这是一个不可忽略的错误,会导致运行程序失败 .一个简单有效的解决方法是:
vim ~/.bashrc
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBARARY_PATH # 添加到最后一行
source ~/.bashrc
- 至此torch 7就配置成功了 .
5. 后记
torch 已经不再更新,取而代之的是 pytorch,不过很多源程序是用torch实现的,配置torch仅仅是为了快速复现和查看效果 .
上述安装过程中90%的问题可能出自网络原因,解决方法是手动下载并安装依赖包 .
# github上查找对应包
git clone package-name.rockspec
luarocks install package-name.rockspec