Yijunmaverick / UniversalStyleTransfer

The source code of NIPS17 'Universal Style Transfer via Feature Transforms'.
MIT License
601 stars 90 forks source link

Requires cuDNN 5 even after downloading encoders& decoders. #12

Open victorkitov opened 5 years ago

victorkitov commented 5 years ago

I downloaded encoders&decoders and replaced decoders with the these decoders (for those not having cuDNN). But actually I have cuDNN7 (cuDNN5 is not supported with my CUDA version 9.1).

Then I run th test_wct.lua -content input/content/04.jpg -style input/style/09.jpg -alpha 0.6

and get the following error:

/home/apogentus/Programs/Torch/install/bin/luajit: ...ntus/Programs/Torch/install/share/lua/5.1/trepl/init.lua:389: ...ntus/Programs/Torch/install/share/lua/5.1/trepl/init.lua:389: ...entus/Programs/Torch/install/share/lua/5.1/cudnn/ffi.lua:1603: 'libcudnn (R5) not found in library path.
Please install CuDNN from https://developer.nvidia.com/cuDNN
Then make sure files named as libcudnn.so.5 or libcudnn.5.dylib are placed in
your library load path (for example /usr/local/lib , or manually add a path to LD_LIBRARY_PATH)

Alternatively, set the path to libcudnn.so.5 or libcudnn.5.dylib
to the environment variable CUDNN_PATH and rerun torch.
For example: export CUDNN_PATH="/usr/local/cuda/lib64/libcudnn.so.5"

stack traceback:
    [C]: in function 'error'
    ...ntus/Programs/Torch/install/share/lua/5.1/trepl/init.lua:389: in function 'require'
    test_wct.lua:9: in main chunk
    [C]: in function 'dofile'
    ...rams/Torch/install/lib/luarocks/rocks/trepl/scm-1/bin/th:150: in main chunk
    [C]: at 0x55e65adf5570

What can I do to run the style transfer?