RuntimeError:Attempting to deserialize object on CUDA devices but torch.cuda.is_available() is False
·
问题
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

运行命令,重新运行程序运行正常
更多推荐
所有评论(0)