Is there any difference between CUDNN_LIB_DIR and CUDNN_LIBRARY enviroment variables fed to the setup.py script ?
The fillowing line suggests that they should be the same, is this true ? If so, why do both of them exist ?
Not sure about these in particular, but I would guess that these two variables are used by two different sub-libraries that are used by pytorch. In particular this makes sure that all of them compile with the same version of cudnn.
Okay, thanks for the reply, but the documentation does not make clear how to set them - should I set them both or just one will suffice. WIll it make sense to open a feature request, requesting to either merge them to one or add more comments to the docs ?
Okay, thanks for the reply, but the documentation does not make clear how to set them - should I set them both or just one will suffice. WIll it make sense to open a feature request, requesting to either merge them to one or add more comments to the docs ?
Hi,
Maybe @Soumith_Chintala can answer to that here (or point it to someone that knows)?
Did you mean @smth ?
just set CUDNN_LIBRARY OR CUDNN_LIB_DIR as environment variable. CUDNN_LIBRARY takes precedence if specified.
2 Likes