Torch not compiled with CUDA enabled,cuda11.6对应的pytorch版本

cuda11.6对应的pytorch版本

Torch not compiled with CUDA enabled问题

这两天运行深度学习的模型一直在报错这个问题,由于本人也是小白一个,在网上找了经验得知“Torch not compiled with CUDA enabled”是因为CUDA和pytorch版本不对应导致。这里提供给大家cuda11.6 的解决办法。

1、搜索pytorch,按下图进行选择(CUDA11.6)
在这里插入图片描述
2、点击上图的Pip,复制Run this Command里的代码到终端里运行即可。
在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
出现AssertionError: Torch not compiled with CUDA enabled错误是因为在编译PyTorch时没有启用CUDA支持。要解决这个问题,你可以按照以下步骤进行操作: 1. 首先,确保你的系统上安装了CUDA驱动程序。你可以从NVIDIA官方网站下载并安装适用于你的显卡的CUDA驱动程序。 2. 接下来,检查你的PyTorch版本是否支持CUDA。你可以通过运行以下代码来检查: ```python import torch print(torch.cuda.is_available()) ``` 如果输出结果为True,表示你的PyTorch支持CUDA。如果输出结果为False,则需要重新安装支持CUDAPyTorch版本。 3. 如果你安装的是CPU版本PyTorch,而不是支持CUDA版本,那么你需要重新安装支持CUDAPyTorch。你可以从PyTorch官方网站下载对应版本,并按照官方文档提供的安装步骤进行操作。 4. 如果你已经安装了支持CUDAPyTorch,但仍然遇到了这个错误,那么可能是因为在安装PyTorch时没有正确设置CUDA相关的环境变量。你可以按照以下步骤进行操作: - 在你的系统环境变量中添加CUDA的路径。具体的路径可能会因为你的系统配置而有所不同,你可以在CUDA安装目录下找到相应的路径。 - 在你的PyCharm项目中设置CUDA相关的环境变量。你可以在PyCharm的设置中找到项目的环境变量设置,并添加CUDA的路径。 通过以上步骤,你应该能够解决AssertionError: Torch not compiled with CUDA enabled错误并成功启用CUDA支持。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值