Running lassonet with CUDA 11.2

lasso-net / lassonet

Feature selection in neural networks

MIT License

207 stars 52 forks source link

Running lassonet with CUDA 11.2 #50

Closed ElrondL closed 11 months ago

ElrondL commented 11 months ago

Hi, is it possible to run GPU lassonet with CUDA 11.2? This is not clear in the documentation. In installation I can see that the packages are trying to install CUDA 12.3, but in my environment I have previously installed CUDA 11.2 via Conda. Does this mean that installing lassonet overwrites system’s CUDA version to 12.3? My driver cannot support 12.3 yet…

louisabraham commented 11 months ago

LassoNet depends on PyTorch. I'm not sure how it works, please look on https://pytorch.org/get-started/locally/