Upgrade PyTorch

Looks like I need to upgrade PyTorch! The error message:

NVIDIA GeForce RTX 3070 with CUDA capability sm_86 is not compatible with the current PyTorch installation. The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70. If you want to use the NVIDIA GeForce RTX 3070 GPU with PyTorch, please check the instructions at

https://pytorch.org/get-started/locally/

Install the latest CUDA version on Windows (3.0GB) – https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=11&target_type=exe_local

Upgrade PyTorch:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

Get arch list in Python.

import torch
torch.cuda.get_arch_list()

[‘sm_37’, ‘sm_50’, ‘sm_60’, ‘sm_61’, ‘sm_70’, ‘sm_75’, ‘sm_80’, ‘sm_86’, ‘compute_37’]

And works!