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

posted @ 2024-02-29 11:01  //toMe  阅读(9)  评论(0编辑  收藏  举报