$CUDAHOME or $CUDA_HOME

kakao / buffalo

TOROS Buffalo: A fast and scalable production-ready open source project for recommender systems

Apache License 2.0

575 stars 106 forks source link

Closed skyer9 closed 5 years ago

skyer9 commented 5 years ago

i think $CUDA_HOME is used more frequently than $CUDAHOME.

ummae commented 5 years ago

It would be better to check both values.

ummae commented 5 years ago

We update the locations where finding CUDA libs as CUDA_PATH, CUDAHOME, CUDA_HOME.