模块'cudnn'未找到

尝试运行https://github.com/rtqichen/style-swap

th style-swap.lua --content images/content/bike.jpg --style images/style/starry_night.jpg
{
  gpu : 0
  saveOriginal : false
  style : "images/style/starry_night.jpg"
  decoder : ""
  tv : 1e-07
  optim : false
  patchSize : 3
  init : "content"
  saveLoss : false
  layer : "relu3_1"
  printEvery : 50
  save : "output"
  learningRate : 0.05
  maxStyleSize : 512
  numSwap : 1
  content : "images/content/bike.jpg"
  maxContentSize : 640
  contentBatch : ""
  cpu : false
  optimIter : 100
  pooling : "max"
  patchStride : 1
}
加载Lua模块...
/usr/bin/luajit: /usr/share/lua/5.1/trepl/init.lua:389: 模块'cudnn'未找到:
    no field package.preload['cudnn']
    no file './cudnn.lua'
    no file '/usr/share/luajit-2.1.0-beta3/cudnn.lua'
    no file '/usr/local/share/lua/5.1/cudnn.lua'
    no file '/usr/local/share/lua/5.1/cudnn/init.lua'
    no file '/usr/share/lua/5.1/cudnn.lua'
    no file '/usr/share/lua/5.1/cudnn/init.lua'
    no file './cudnn.so'
    no file '/usr/local/lib/lua/5.1/cudnn.so'
    no file '/usr/lib/x86_64-linux-gnu/lua/5.1/cudnn.so'
    no file '/usr/local/lib/lua/5.1/loadall.so'
stack traceback:
    [C]: in function 'error'
    /usr/share/lua/5.1/trepl/init.lua:389: in function 'require'
    style-swap.lua:65: in main chunk
    [C]: in function 'dofile'
    /usr/lib/torch-trepl/th:149: in main chunk
    [C]: at 0x5587020201d0

如何安装cudnn?

sudo luarocks install cudnn

错误:未找到与查询匹配的结果。

sudo luarocks install cutorch

错误:未找到与查询匹配的结果。

sudo luarocks install cunn

错误:未找到与查询匹配的结果。
点赞
用户1179925
用户1179925

似乎 torch 有它自己的版本的 luarocks https://github.com/szagoruyko/loadcaffe/issues/48#issuecomment-189981690

2019-11-04 19:28:30