我先说明,我是个新手,我只是把困扰了我很久的问题解决了,然后写个记录。
问题描述
我最开始按照Bubbliiiing大佬分享的博客内容进行目标检测相关的学习,原来用的是1650的笔记本,一切顺利,唯独显存不足,有些网络跑不起来。
后来我换了一台台式电脑,显卡是RTX 3060的,重新按照教程配置环境总会出各种乱七八糟的问题,什么都有,总结一句话就是代码跑不起来。原来笔记本都能运行的代码,现在台式机都不能用,一时间我很沮丧。
总之,网上各种找解决办法。
解决办法
直接说结论吧,原本按照上面链接我应该是配置torch1.2.0和torchvision0.4.0的,但是我在评论区中看到大佬说30系显卡得用cuda11和torch1.8,因此恍然大悟,环境配置上我需要修改。
我搜索了很久,找到了一个网页,链接如下,他提供了一个适合cuda11的torch1.8和torchvision0.9安装链接,我之前查过,torch1.8需