ランタイムエラーの解決方法

RuntimeError                              Traceback (most recent call last)
<ipython-input-10-e63a24435863> in <cell line: 2>()
      1 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
----> 2 weigths = torch.load('yolov7-w6-pose.pt')
      3 model = weigths['model']
      4 model = model.half().to(device)
      5 _ = model.eval()

6 frames
/usr/local/lib/python3.10/dist-packages/torch/serialization.py in validate_cuda_device(location)
    256 
    257     if not torch.cuda.is_available():
--> 258         raise RuntimeError('Attempting to deserialize object on a CUDA '
    259                            'device but torch.cuda.is_available() is False. '
    260                            'If you are running on a CPU-only machine, '

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
weigths = torch.load('yolov7-w6-pose.pt')
model = weigths['model']
model = model.half().to(device)
_ = model.eval(