devenv icon indicating copy to clipboard operation
devenv copied to clipboard

Published 20 hours ago • verified publisher icon cachix

cuda: add module

Open bobvanderlinden opened this issue 1 year ago • 15 comments

As discussed on Discord, this configuration is needed to run pytorch in devenv on Linux. It was confirmed to work.

I don't have much knowledge of CUDA itself, so I'm unsure what other libraries exactly need. I did find that CUDA_HOME and CUDA_PATH are used by tensorflow.

Confirmations that this works for real projects are welcome!

bobvanderlinden avatar Feb 20 '23 20:02 bobvanderlinden