kingbri 1e7cf1e5a4 Start: Prompt user for GPU/lib ...

There is no platform agnostic way to fetch CUDA/ROCm's versions
since environment variables change and users don't necessarily need
CUDA or ROCm installed to run pytorch (pytorch installs the necessary
libs if they don't exist).

Therefore, prompt the user for their GPU lib and store the result in
a textfile so the user doesn't need to constantly enter a preference.

Signed-off-by: kingbri <bdashore3@proton.me>

2024-03-20 15:21:37 -04:00

4.2 KiB

Raw Blame History

View Raw