CycleGAN icon indicating copy to clipboard operation
CycleGAN copied to clipboard

cuDNN lib error when selected CPU

Open khemendra0 opened this issue 3 years ago • 0 comments

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

khemendra0 avatar May 06 '21 03:05 khemendra0