配置 Torch 7

参考博客 :


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
  • 配置成功会看到类似下图的版本信息 .
    cuda和cudnn版本信息

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
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值