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