目录

1.CUDA下载安装步骤

2. Pytorch环境的配置

笔者计算机视觉研0刚入学为研一。近期在学习目标检测算法中的YOLO系列。在运行YOLO V1训练代码时,报出该错误

原因很简单:CUDA和Torch版本不兼容

遇到这类问题先检查电脑的CUDA支持版本:

打开cmd,输入

nvidia-smi

可以看到红框里的是电脑支持的最高版本的cuda,我们在官网进行下载时,下载该版本及以下的即可(建议下载该版本以下)

其次,检查自己的电脑中CUDA的版本(检查是否安装CUDA):

打开cmd,输入

nvcc --version

可以看到我最开始的安装版本是11.7

如果显示无法找到nvcc说明电脑中没有安装CUDA。

补充:

如果你的电脑中有CUDA但版本不合适,删除重装即可,需要注意的是,一定要卸载干净。我是在电脑自带的控制面板中进行卸载的:点击控制面板——程序——程序和功能

找到所有和NVIDIA有关的全部卸载,然后再去软件管家清理卸载残留。其他的卸载方式也可,但一定要卸载干净。

1.CUDA下载安装步骤

如果电脑支持CUDA11.x,最好下载11.3版本,这样比较保险。

点击CUDA官网:https://developer.nvidia.com/cuda-downloads

选择更多之前版本:

根据自己电脑的系统选择合适的进行下载即可:

我是把它放到了F盘中:

安装包下载好后,右键——管理员运行——进入安装,无脑点下一步即可(尽量按照系统为你选择好的安装路径来安,也就是C盘路径。之前我尝试在其他盘中安装,没有成功),需要注意:在“安装选项”这一栏选择:自定义安装。

安装完成后,检查是否安装成功,再次打开cmd,输入

nvcc --version

可以看到我现在的版本变成了11.3,说明安装成功。

2. Pytorch环境的配置

首先对照CUDA和pyTorch对应的版本,可以看到CUDA对应的pytorch版本有三个,这里我选择1.11.0

选择好版本后我们就开始下载pytorch安装包。

方法有两种:

1.在pytorch官网:https://pytorch.org/get-started/locally/选择在线安装:

但这样的安装方式需要极好的网络环境,中间过程出错率较高。我采用的是第二种方法,离线下载。

2.离线安装:http://2.https://download.pytorch.org/whl/torch_stable.html

选择对应的版本即可,这里的cu:cuda版本;cp:Python版本;torch:torch版本,选择适合自己的系统,点击链接下载即可。

Torch下载好后,进行离线安装:

打开Anaconda

切换安装路径:直接复制路径是切不过来的(切换路径需要输入cd)

需要在此基础上直接输入:

F:

回车,即可转到F盘路径下

我们需要在这个路径下安装torch。在命令行输入

pip install torch

加上一个Tab键,导出torch安装链接,并在后面加上清华大学的镜像源:https://pypi.tuna.tsinghua.edu.cn/simple/加快安装速度

如果报错:

显示pip需要更新,可以直接去cmd中输入:

python -m pip install --upgrade pip

更新即可

随后等待torch的安装,显示成功即安装完成

安装完成后,我们可以进行简单的版本测试:

在anaconda中输入:

import torchprint(torch.__version__):查看torch版本print(torch.cuda.is_available()):看安装好的torch和cuda能不能用,也就是看GPU能不能用

返回如下图所示,即代表安装成功

回到train训练脚本中,运行代码:

Bug解决!

希望可以帮到各位卷王~~

报错:Torch not compiled with CUDA enabled看这一篇就足够了相关推荐

  1. AssertionError: Torch not compiled with CUDA enabled【pycharm/python3/pip】

    目录 问题 查阅资料 解决步骤 1.检查pytorch版本.是否有CUDA 2.安装CUDA前看电脑的显卡驱动程序版本.支持的最高版本 3.安装CUDA和cuDNN 4.卸载pytorch 5.重新安 ...

  2. 报错 AssertionError: Torch not compiled with CUDA enabled

    报错: AssertionError: Torch not compiled with CUDA enabled 本文针对仅有CPU无GPU的设备,针对该报错进行修改: 在代码开头添加: // A c ...

  3. Torch not compiled with CUDA enabled 报错的归纳总结

    以前总是嫌装环境太麻烦,碰到些需要用到GPU的项目都不想去复现了...这次因为论文需要,下定决心要把pytorch的安装问题搞定,但是期间遇到了很多问题,最烦人的莫过于这个' AssertionErr ...

  4. AssertionError: Torch not compiled with CUDA enabled到底是什么原因

    代码运行报错AssertionError: Torch not compiled with CUDA enabled 但是根据cuda安装显示 cuda是安装成功的 这是怎么回事 我先进入百度翻译 A ...

  5. AssertionError: Torch not compiled with CUDA enabled 解决方案

    报错 Traceback (most recent call last):File "main_pretrain.py", line 250, in <module>m ...

  6. 解决 AssertionError Torch not compiled with CUDA enabled

    最近在矩池云的的Tesla K80机子上跑MMYOLO,跟着MMYOLO官方文档<自定义数据集 标注+训练+测试+部署 全流程 >操作到 "2.1.1 软件或者算法辅助" ...

  7. Torch not compiled with CUDA enabled

    Torch not compiled with CUDA enabled 如果没有可用的显卡资源,则检查代码,把调用显卡资源的部分全部去掉即可 https://blog.csdn.net/qq_403 ...

  8. 成功解决raise AssertionError(“Torch not compiled with CUDA enabled“)AssertionError: Torch not compiled

    成功解决raise AssertionError("Torch not compiled with CUDA enabled") AssertionError: Torch not ...

  9. Pytorch错误:Torch not compiled with CUDA enabled

    一.问题 今天跑一个项目代码遇到了一个问题,大致是pytorch和CUDA版本不匹配,然后我就降低pytorch版本,结果1.7中网络不包含我所需的函数,我直接pip install pytorch= ...

最新文章

  1. (C++)CSP 201712-2 游戏
  2. 马斯克公开支持“上班摸鱼”:让工作更愉快!
  3. 不是我放弃AI,是AI放弃了我!!
  4. 人人都是产品经理 2
  5. linux epoll事件模型详解
  6. python字符编码
  7. docker 安装入门
  8. 计算机网络严楠,安徽工程大学
  9. (一)选择排序之一:堆排序
  10. Dubbo面试 - dubbo的工作原理
  11. 华为Mate 30系列后摄相机模组曝光:圆形三摄各有不同
  12. linux开源镜像站 目录帮助,LinuxEye开源镜像站运营
  13. 【C语言开源项目】盘点 GitHub 上不错的 4 个C语言项目
  14. 珠心算 详解(C++)
  15. 好有成就感。。。又编了一个扫雷
  16. 电路中的正弦信号 Sin
  17. jmeter做秒杀活动测试
  18. 有用但用处不多的html的属性
  19. apple pay,--牛逼,
  20. 惠普m154a状态页_惠普M154a说明书

热门文章

  1. python自动输入文字_Python自动输入【新手必学】
  2. php获取微博热搜,爬取微博热搜top50(示例代码)
  3. 【IoT】物联网NB-IoT之电信物联网开放平台对接流程浅析
  4. matlab中contourm,MATLAB 中contour函数的使用
  5. 什么行业需要用到高防服务器呢?高防服务器又有什么用?
  6. 【Luogu P1151】子数整数
  7. Tracup|远程办公,为什么会让管理者抓狂?
  8. 对 kubeadm 进行故障排查
  9. css实现固定宽高比例的div
  10. 传iPhone8将搭载裸眼3D技术