I used
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
the torch library is working, if I just use device=cpu instead of device=cuda, then I don’t get any error
import torch

print(torch.__version__)
my_tensor = torch.tensor([[1, 2, 3], [4, 5, 6]], dtype=torch.float32, device="cpu")
print(my_tensor)
torch.cuda.is_available()