问题

raise RuntimeError('Attempting to deserialize object on a CUDA ’
RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False.
在这里插入图片描述
当前环境不支持运行CUDA

解决

1、在环境中查看是否安装pytorch

pip list

2、在激活的环境中输入python
输入以下命令

import torch
torch.__version__

此时显示pytorch版本是cpu版本,不支持gpu,需要在官网重新安装下载
在这里插入图片描述

3、查看cuda版本

nvidia-smi

在这里插入图片描述
4、到Pytorch官网选择适合的版本,pytorch官网:https://pytorch.org/get-started/locally/

复制命令进行安装

pip install torch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 --index-url https://download.pytorch.org/whl/cu121

在这里插入图片描述
运行命令,重新运行程序运行正常

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐