Ensure correct path is used for importing math_functions

NVIDIA / FastPhotoStyle

Style transfer, deep learning, feature transform

Other

11.14k stars 1.2k forks source link

Ensure correct path is used for importing math_functions #76

Closed arijoon closed 3 years ago

arijoon commented 5 years ago

The include path for postprocessing is not correct if this is not ran on a linux machine and have cuda in a very specific directory which renders setting C_INCLUDE_PATH pointless and causes this error:

pynvrtc.compiler.ProgramException: best_local_affine_kernel.cu(2): catastrophic error: cannot open source file "/usr/local/cuda/include/math_functions.h"