当前位置:   article > 正文

Torch安装及使用_安装torch

安装torch

安装 Torch

  • 系统:Ubuntu 12+

依次执行下面四条语句:

# in a terminal, run the commands WITHOUT sudo
git clone https://github.com/torch/distro.git ~/torch --recursive

cd ~/torch

bash install-deps

bash install.sh
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8

说明: 第一条语句的默认安装路径在 home/usr/ 下,根据需要可以修改

  • 第 1 条语句首先安装 LuaJIT 和 Torch 所需的基本包依赖项
  • 第 3 条语句安装 LuaJIT、LuaRocks,然后使用 LuaRocks (Lua包管理器) 安装核心包,如 TORCH、nn 和 PATH,以及一些其他软件包。

脚本将 Torch 添加到 PATH 变量中。只需 source 一次就可以刷新环境变量。安装脚本将检测到当前的 shell,并在正确的配置文件中修改路径。

刷新方法:

# On Linux with bash
source ~/.bashrc
# On Linux with zsh
source ~/.zshrc
# On OSX or in Linux with none of the above.
source ~/.profile
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6

删除 Torch

rm -rf ~/torch
  • 1

安装包

使用 luarocks(包管理工具) 安装新的包:

$ # run luarocks WITHOUT sudo
$ luarocks install image
$ luarocks list
  • 1
  • 2
  • 3

进入和退出 Torch

$ th

 ______             __   |  Torch7                                   
 /_  __/__  ________/ /   |  Scientific computing for Lua.         
  / / / _ \/ __/ __/ _ \  |                                           
 /_/  \___/_/  \__/_//_/  |  https://github.com/torch   
                          |  http://torch.ch            
			  
th> torch.Tensor{1,2,3}
 1
 2
 3
[torch.DoubleTensor of dimension 3]

th>
th> os.exit()
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
  • 10
  • 11
  • 12
  • 13
  • 14
  • 15
  • 16

遇到的问题

error: more than one operator “==” matches these operands

执行第四条语句容易报错:error: more than one operator “==” matches these operands”,是因为 cuda 和 torch 的头文件都提供了相同的重载运算符,这样编译器在使用的时候就会出现混乱。
解决方法:禁止使用cuda的头文件编译torch

export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"
  • 1

安装完毕后需要先重启终端,再输入 th,如果没有报错说明安装成功。
在这里插入图片描述

luarocks install cutorch 出错

luarocks install cutorch 出错,要保证 luarocks 调用的是~/torch/install/bin/ 下的 luarocks,且用 sudo 执行。

sudo /data/monn/torch/install/bin/luarocks install cutorch
  • 1

Error: No results matching query were found.

在使用luarocks install时,莫名其妙的提示了

warning: Failed searching manifest: Failed fetching manifest for …
Error: No results matching query were found.
  • 1
  • 2

解决方法:删除缓存文件,在终端输入:

sudo rm -rf ~/.cache/luarocks
  • 1

或者在linux的文件管理器中显示隐藏文件,然后在“/home/当前用户”下找到.cache文件夹,进去后里面有luarocks文件夹,删除该文件夹(或者使用sudo rm -rf),之后重新 luarocks install 即可。

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Gausst松鼠会/article/detail/308579
推荐阅读
相关标签
  

闽ICP备14008679号