30系显卡配置pytorch环境

本文记录了作者在使用RTX 3060显卡配置PyTorch环境时遇到的问题及解决办法。原本按照教程配置torch1.2.0和torchvision0.4.0出现错误,通过查找资料发现30系显卡需搭配cuda11和torch1.8。遵循新的环境配置指南,安装cuda11.1和cudnn11.1,成功运行代码。最后提供了验证GPU是否可用的代码片段。
摘要由CSDN通过智能技术生成

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

问题描述

我最开始按照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需

  • 15
    点赞
  • 21
    收藏
    觉得还不错? 一键收藏
  • 8
    评论
评论 8
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值