NVIDIA-Merlin / core

Core Utilities for NVIDIA Merlin
Apache License 2.0
19 stars 14 forks source link

Update HAS_GPU variable to account for `CUDA_VISIBLE_DEVICES` #221

Closed oliverholworthy closed 1 year ago

oliverholworthy commented 1 year ago

Fixes #220

Update HAS_GPU variable to account for the CUDA_VISIBLE_DEVICES environment variable.

This is to handle the case where this env var is set to an empty string CUDA_VISIBLE_DEVICES="". In this case RAPIDS cudf and other libaries that use CUDA will not detect a GPU.