Torch not compiled with CUDA enabled 解决办法

## 目标

解决Torch not compiled with CUDA enabled 版本不兼容问题

### 原理
Torch not compiled with CUDA enabled,即CUDA、torch的版本兼容问题是造成print(torch.cuda.is_available()) 返回false的一种原因。本文搜集从卸载CUDA、torch,选择兼容版本,再到使用清华源,最后安装CUDA、torch的博客集合,作为解决方法:
如何查看torch版本:https://blog.csdn.net/weixin_47414034/article/details/124632748
如何查看cuda版本:https://blog.csdn.net/weixin_38723958/article/details/123143521

卸载torch:https://blog.csdn.net/zaf0516/article/details/103927989
卸载CUDA:https://blog.csdn.net/Zinnir/article/details/122766367

安装 PyTorch 切换清华源镜像:https://blog.csdn.net/weixin_44589540/article/details/122847278

Python配置清华镜像源_视觉闫小亘的博客-CSDN博客_python清华镜像源
CUDA历史版本下载:https://developer.nvidia.com/cuda-toolkit-archive
PYTORCH历史版本下载:https://pytorch.org/get-started/previous-versions/
PYTORCH版本建议按照网页上的CUDA版本匹配下载对应的版本

conda创建新环境:conda创建新环境_joshuwang0810的博客-CSDN博客_conda create
conda 多环境切换:https://blog.csdn.net/yuehenmiss/article/details/102914941

最后用python验证安装情况,如果返回True,则成功;否则,用conda list检查pytorch是否为CPU版本,再尝试安装CUDA版本的pytorch

import torch
 
print(torch.__version__)#torch版本
 
print(torch.cuda.is_available())#GPU是否可用

感谢以上博文的作者

  • 6
    点赞
  • 46
    收藏
    觉得还不错? 一键收藏
  • 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,则需要重新安装支持CUDA的PyTorch版本。 3. 如果你安装的是CPU版本的PyTorch,而不是支持CUDA的版本,那么你需要重新安装支持CUDA的PyTorch。你可以从PyTorch官方网站下载对应的版本,并按照官方文档提供的安装步骤进行操作。 4. 如果你已经安装了支持CUDA的PyTorch,但仍然遇到了这个错误,那么可能是因为在安装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、付费专栏及课程。

余额充值