What is the command to install pytorch with cuda 12. 8? 1 as of now, pytorch which supports cuda 12 8 is not released yet but unofficial support released nightly version of it here are the commands to install it so with this pytorch version you can use it on rtx 50XX I've got 5080 and it works just fine
gpu - Pytorch version for cuda 12. 2 - Stack Overflow Pytorch version 1 5 works with Pytorch for CUDA 10 2 However, the problem I have is it seems Anaconda keeps downloading the CPU libaries in Pytorch rather than the GPU