Hello,
I actually realized (way too late) that it was because pip kept installing torch from cached, and so kept installing the cpu only version (only confirmed when I went into torch/version.py). In any case, it still didn’t work in my normal environment, but worked when I ran it in a new fresh conda install. Thank you!