[NEED HELP] Trouble with CUDA capability sm_86

Thank you very much. I had the same problem, and it works well right now. I reinstalled pytorch with CUDA 11.7 in Anaconda from website

1 Like

Hi,

I have similar problems with PyTorch support for sm_86. I have a NVIDIA A4500 GPU with 12.1 CUDA on ubuntu 20.04 LTS. Is there perhaps a simple tool to make the right selection of GPU, CUDA and PyTorch version - I do not want to build PyTorch from source. If not please can any one help with installation script for this. Would really appreciate it.

You can go to the linked website, select the desired CUDA 11.x version (or 12.1 for the nightlies), copy/paste the install command, and execute it, since all of our binaries support sm_86.

Sorry Patrick,

can you provide the link -really appreciate it!

Kind regards,

Alfons.

Sure, here it is again: Start Locally | PyTorch