解决报错:AssertionError: Torch not compiled with CUDA enabled

  1. 首先查看自己的cuda是否可用
torch.cuda.is_available()

在这里插入图片描述
这里我的cuda是不适配torch的,所以需要重新安装适配的torch

  1. 查看自己的cuda版本
    方法1
    在这里插入图片描述
    方法2
    在cmd处输入nvidia-smi
    在这里插入图片描述
    这样可以找到的自己的CUDA版本
  2. 安装符合自己版本的pytorch
    进入pytorch官网https://pytorch.org/get-started/locally/安装自己需要的pytorch
    这里都是最新版的Pytorch,如何没有你需要的版本,可以安装之前版本的pytorch
    在这里插入图片描述

因为我的cuda版本为11.7,这里我就安装的pytorch1.13.1版本的
在这里插入图片描述
在这里插入图片描述

pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117

安装成功
在这里插入图片描述
查看这时cuda是否可用
在这里插入图片描述
这里显示为true,完美解决,撒花

注:在这个过程中,我还尝试了安装torch2.0.0和2.0.1版本,但是都不可用,显示的是cpu版本的,我在查阅问题原因,说的是cuda11.7不接受2.0以上的版本,建议各位安装2.0以下的版本。

  • 9
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值