Windows下配置yolov5环境(显卡RTX NVIDIA3070TI)

本文详细介绍了在Windows系统中,如何为RTX NVIDIA3070TI显卡配置CUDA和cuDNN,以支持深度学习框架pytorch。首先,查看显卡支持的CUDA最高版本,然后下载并安装CUDA,注意避免设置相同的临时解压和安装路径。接着,安装cuDNN,将文件复制到CUDA安装目录。最后,安装与CUDA版本匹配的pytorch,并验证CUDA和torch的版本。
摘要由CSDN通过智能技术生成

NVIDIA显卡:是一款硬件板子,主流是NVIDIA的GPU,深度学习本身需要大量计算。GPU的并行计算能力,在过去几年里恰当地满足了深度学习的需求。
显卡驱动:告诉计算机识别显卡硬件,调用其计算资源。
CUDA:是NVIDIA推出的只能用于自家GPU的并行计算框架。只有安装这个框架才能够进行复杂的并行计算。主流的深度学习框架也都是基于CUDA进行GPU并行加速的。

cudnn:是针对深度卷积神经网络的加速库。
————————————————
(原文链接:https://blog.csdn.net/jyfhaoshuai/article/details/118640470)

一:查看自己显卡驱动可以支持的CUDA版本的最高版本(注意:RTX30系列显卡只能支持CUDA11以上的版本!)

(通常新版本的显卡驱动应该能支持当前最高支持CUDA版本以下的所有版本,所以要关注显卡驱动能支持的最高CUDA版本。)

1.打开电脑左下角【搜索键】,搜索NVIDIA控制面板,并打开:

2.点击设置Physx设

  • 1
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值