yolo7检测学习Bubbliiiing的视频有感——(3)AssertionError: Torch not compiled with CUDA enablej
出现这个问题的主要原因在于pytorch和cuda版本不兼容
由于我是完全按照B导教程安装的,也检查过自己的显卡系列,所以对这个报错持怀疑态度
1.先百度确认自己的显卡信息,确认无误,比如AMD显卡只能用cpu进行训练,我这里的显卡用到的各个软件版本信息为
torch 1.2.0
cuda 10.0
ptyhon 3.6.10
注意各个版本必须要相互对应的,有些安装的太高会导致不兼容问题发生
使用pip list查看自己安装的库
可以看到我这里是已经安装了torch1.2.0且检查了其他软件信息,版本号也能互相对应
2.安装网上的信息检查当前导入项目的torch版本
import torch
print(torch.__version__)
打印信息版本为1.13.1
从而确认当前环境存在多个pytoch,且我使用的是1.13.1版本,到这里已经很清晰了需要明确自己是否激活了自己的虚拟环境,不同环境下的torch使用版本不一样
1.13.1是我电脑自带的cuda,然后仔细看发现我没有激活环境,用conda activate xxxx激活自己的虚拟环境重新运行就没有报错了
3.后续重新运行又报错ModuleNotFoundError: No module named 'tensorboard'
查看自己是否安装tendorboard
tensorboard --version
没有安装继续安装就行了,安装后用命令激活
tensorboard --logdir=./logs
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· Manus重磅发布:全球首款通用AI代理技术深度解析与实战指南
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· 【自荐】一款简洁、开源的在线白板工具 Drawnix