logoalt Hacker News

tomtom1337 • today at 7:07 AM • 0 replies • view on HN

One criticism is that you have to install the same package, torch, but from different Python indexes in order to install the cpu version or gpu version, on Linux. On windows, `pip install torch` gets you the cpu version. On linux, that gets you a ton of Nvidia extras that take a lot of space.

GPU support should really be a optional extra eg `torch[gpu]` or `torch[nvidia]`.