/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/utils/cuda_enabled.h (154B)
#pragma once
namespace torch {
namespace utils {
static inline bool cuda_enabled() {
#ifdef USE_CUDA
return true;
#else
return false;
#endif
}
}
}