bubbliiiing/centernet-pytorch

按视频操作CUDA和pytorch不匹配

kimtaesa opened this issue · 0 comments

博主你好,我在跟视频操作的时候发现运行predict文件报错图片如下
图片1
我按照视频操作安装的torch和torchvision 和cuda以及cudnn,但是运行predict时报错是AssertionError: Torch not compiled with CUDA enabled,应该怎么办博主