Add possibility of installing CPU version only

StefanDevAccount / LearnMHN

A Python package to work with Mutual Hazard Networks (MHN)

MIT License

0 stars 2 forks source link

Add possibility of installing CPU version only #2

Closed pawel-czyz closed 1 year ago

pawel-czyz commented 1 year ago

Hi, I like the CUDA support, but sometimes it is quite tricky to install the right version of drivers (especially when working on clusters with different GPUs).

Hence, I'd like to be able to specify that CUDA should not be used at the installation stage. One way of doing this is via environmental variables, as pip install mhn command does not allow one to specify additional boolean flags.

Please, let me know what you think!

StefanDevAccount commented 1 year ago

Hey, really cool feature! I will merge your changes into the package.