Cuda/GPU incompatibilities

[stav-ros]

Not necessarily an issue, but for those with newer GPUs (4090 in my case) requiring newer CUDA, this worked for me:

pip install torch==2.0.1+cu118 torchvision==0.15.2+cu118 torchaudio --extra-index-url https://download.pytorch.org/whl/cu118

You also might need to run
pip install numpy numba scipy --upgrade
(Tutorial) Code runs like butter.

[asarnow]

I also had to set CUDA_VISIBLE_DEVICES, and the above was working.