Same as here. Your locally installed CUDA toolkit won’t be used and you need to install a PyTorch binary with CUDA>=11.x.