La forma más “Pythonic” de tener un paquete que instale dependencias en función de la presencia o fabricante de la GPU es la siguiente:

Actualmente estoy desarrollando una biblioteca de Aprendizaje Automático que permite a los usuarios escribir código independiente de CPU y GPU para lograr ciertas tareas. Para ejecutar el código habilitado para GPU, mi paquete tiene dependencias específicas que solo son compatibles con GPU de NVIDIA habilitados para CUDA. Por lo tanto, me gustaría que mi paquete solo instale esas dependencias cuando un usuario instale mi paquete en un dispositivo con GPU habilitado para CUDA. Investigué cómo hacer esto y al principio parecía que la respuesta estaba en PEP 508. Sin embargo, los “marcadores de entorno” descritos en este PEP no incluyen la presencia de una GPU, y mucho menos el fabricante y/o controladores de esa GPU.

Por lo tanto, estoy buscando la solución más “Pythonic” para instalar ciertas dependencias en función de la presencia de una GPU de NVIDIA habilitada para CUDA (si es mucho más fácil simplemente comprobar la presencia de una GPU de NVIDIA sin verificar el estado de CUDA, puedo hacerlo como solución por el momento, pero definitivamente preferiría verificar el estado habilitado para CUDA).

Entiendo que setup.py puede ejecutar código Python arbitrario, por lo que técnicamente puedo resolver este problema con código dentro de ese archivo que lo haga por mí. Sin embargo, tengo entendido que la comunidad de Python se está alejando de la dependencia de setup.py debido a los riesgos inherentes asociados con la ejecución de código arbitrario durante la instalación del paquete. Por lo tanto, me gustaría cumplir con las preferencias de la comunidad y que mi solución utilice pyproject.toml si es posible, aunque también estaría dispuesto a usar setup.cfg si se considera más apropiado (para ser honesto, no siento que comprenda completamente la distinción entre estos dos archivos y cuál es más apropiado para qué propósito, aunque según el video de mCoding que vi para configurar mis pruebas parece que setup.cfg se utiliza más para metadatos del paquete).

Aunque lo que estoy buscando hacer no se detalla expresamente en PEP 508, parece que va en el mismo espíritu, por lo que me sorprendería si no hubiera una forma canónica de hacerlo.