Comment

Comment by tomtom1337

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]`.

parent tomtom1337 1 month ago

Discussion

Live from the official Hacker News API

A modern reader for Hacker News · SvelteKit + Skeleton