30系显卡配置pytorch环境

我先说明，我是个新手，我只是把困扰了我很久的问题解决了，然后写个记录。

问题描述

我最开始按照Bubbliiiing大佬分享的博客内容进行目标检测相关的学习，原来用的是1650的笔记本，一切顺利，唯独显存不足，有些网络跑不起来。

参考内容windows下的torch=1.2.0环境配置

后来我换了一台台式电脑，显卡是RTX 3060的，重新按照教程配置环境总会出各种乱七八糟的问题，什么都有，总结一句话就是代码跑不起来。原来笔记本都能运行的代码，现在台式机都不能用，一时间我很沮丧。

总之，网上各种找解决办法。

解决办法

直接说结论吧，原本按照上面链接我应该是配置torch1.2.0和torchvision0.4.0的，但是我在评论区中看到大佬说30系显卡得用cuda11和torch1.8，因此恍然大悟，环境配置上我需要修改。

 我搜索了很久，找到了一个网页，链接如下，他提供了一个适合cuda11的torch1.8和torchvision0.9安装链接，我之前查过，torch1.8需要和torchvision0.9配合使用。

网页链接：https://pytorch.org/get-started/previous-versions/ 

 之后按照Bubbliiiing大佬的博客内容流程继续重头开始配置环境，在安装cuda时则改成11.0版本，打开cmd安装torch与torchvision版本时换成上述截图内容，其余部分可以完全不修改，代码是完全可以运行的。这里提供我自己能用的cuda11.1和cudnn11.1版本：

cuda11.1

cudnn11.1

小结：

安装上面给出的cuda和cudnn，创建pytorch环境则复制图中标注部分代码在cmd里面操作

查看是否可以调用GPU：win+R输入cmd并回车，activate pytorch激活环境，

python

import torch

print(torch.cuda.is_available())

 如果输出是True，则说明环境配置成功，可以调用GPU，程序运行的时候也就不会报错了。

总结

虽然内容不多，但是我找了很久查到的解决方法，记录一下帮助后来人。

如果没有帮到你，请继续百度，我也不懂。第一次写文章，见谅。