search

junyanz / CycleGAN

Software that can generate photos from paintings, turn horses into zebras, perform style transfer, and more.
Other
12.22k stars 1.94k forks source link

cuDNN lib error when selected CPU #145

Open khemendra0 opened 3 years ago

khemendra0 commented 3 years ago

Hi,

I am doing test with CPU and provide option as gpu=0. I am testing with CPU (No gpu) and so added option gpu=0.

but when i start test it failed with cuDNN lib. I was expecting if select cpu, then it should not use cuDNN as cudnn is part of GPU.

root@83821d00459d:~/torch/CycleGAN# DATA_ROOT=./datasets/ae_photos name=style_cezanne_cpu_pretrained model=one_direction_test phase=test gpu=0 loadSize=256 fineSize=256 resize_or_crop="scale_width" th test.lua

loading previously trained model (/root/torch/CycleGAN/checkpoints/style_cezanne_cpu_pretrained/latest_net_G.t7)
use InstanceNormalization
/root/torch/install/bin/luajit: /root/torch/install/share/lua/5.1/trepl/init.lua:389: module 'cudnn' not found:No LuaRocks module found for cudnn no field package.preload['cudnn'] no file '/root/.luarocks/share/lua/5.1/cudnn.lua' no file '/root/.luarocks/share/lua/5.1/cudnn/init.lua' no file '/root/torch/install/share/lua/5.1/cudnn.lua' no file '/root/torch/install/share/lua/5.1/cudnn/init.lua' no file './cudnn.lua' no file '/root/torch/install/share/luajit-2.1.0-beta1/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 '/root/.luarocks/lib/lua/5.1/cudnn.so' no file '/root/torch/install/lib/lua/5.1/cudnn.so' no file '/root/torch/install/lib/cudnn.so' no file './cudnn.so' no file '/usr/local/lib/lua/5.1/cudnn.so' no file '/usr/local/lib/lua/5.1/loadall.so' stack traceback: [C]: in function 'error' /root/torch/install/share/lua/5.1/trepl/init.lua:389: in function 'require' /root/torch/CycleGAN/util/util.lua:189: in function 'load_test_model' ./models/one_direction_test_model.lua:24: in function 'Initialize' test.lua:79: in main chunk [C]: in function 'dofile' /root/torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk [C]: at 0x564c856a2570 root@83821d00459d:~/torch/CycleGAN# DATA_ROOT=./datasets/ae_photos name=style_cezanne_cpu_pretrained model=one_direction_test phase=test gpu=0 loadSize=256 fineSize=256 resize_or_crop="scale_width" th test.lua