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.
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/cu118You also might need to run
pip install numpy numba scipy --upgrade(Tutorial) Code runs like butter.