multiple gpus issuse

[shenglunch]

How to solve this issuse?

File "/public/home/network/depth_pro/src/depth_pro/network/encoder.py", line 275, in forward
x_pyramid_encodings = self.patch_encoder(x_pyramid_patches)
File "/public/home/csl/anaconda3/envs/large/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/public/home/anaconda3/envs/large/lib/python3.8/site-packages/timm/models/vision_transformer.py", line 827, in forward_features
x = self.patch_embed(x)
File "/public/home/anaconda3/envs/large/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/public/home/anaconda3/envs/large/lib/python3.8/site-packages/timm/layers/patch_embed.py", line 131, in forward
x = self.proj(x)
File "/public/home/anaconda3/envs/large/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/public/home/anaconda3/envs/large/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 463, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/public/home/anaconda3/envs/large/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 459, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:1 and cuda:0! (when checking argument for argument weight in method wrapper_CUDA__cudnn_convolution)