Hi, everyone. The configuration I use is Tesla A100, cuda11.0, cudnn8.0.4,pytorch1.7.1. But when I run the following code：
import torch
a = torch.randn(2, 3, device='cuda:0')
print(a)
b = a.to('cuda:1')
print(b)

The output are:
tensor([[-2.0747, -0.5964, -0.3556],
        [-0.5845,  0.3389, -0.3725]], device='cuda:0')

tensor([[0., 0., 0.],
        [0., 0., 0.]], device='cuda:1')

I am now trying to confirm if this problem is caused at the code level (pytroch,cuda,cudnn) or if there is something wrong with the gpu installation. Thanks!