conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch if you're using anaconda, you can use this command