Win10配置yolov7环境(pytorch1.7.1 cuda cudnn)

本文详细介绍了在Windows系统上安装CUDA和CuDNN,以及如何创建和激活Python环境来安装PyTorch-GPU。过程中提到了检查NVIDIA显卡驱动和CUDA版本的匹配,安装Anaconda,设置环境变量,以及解决安装过程中可能出现的问题。此外,还涵盖了安装VSCode的步骤,以备代码编辑之需。

 显卡不支持CUDA的不用浪费时间

目录

1.Anaconda安装

2.安装cuda cudnn

3.激活pytorch-gpu环境

4.其他依赖环境安装

5.安装VsCode(可选)

可能的报错

参考博客

1.Anaconda安装

 登录Anaconda的官网:Platform | Anaconda。下载新版安装包

 双击安装,没什么需要注意的。可以不安装在C盘,路径改自己的。

如果有Add Anaconda to my PATH environment variable这一项可以勾选,这样会自动将anaconda装到系统的环境变量中。

如果没有这一项,那么手动添加环境变量,比如我的安装路径是

D:\ProgramData\anaconda3\condabin

2.安装cuda cudnn

首先需要你的显卡支持CUDA,如果你是Nvidia显卡,建议更新下新版显卡驱动,怎么下载安装显卡驱动这里不说了,教程太多了。

可以win+R输入cmd打开命令窗口,输入nvidia-smi

 如图,我最高支持CUDA12.1。其实驱动版本与cuda版本的对应关系官网有,可以查看下面链接:
CUDA 12.1 Update 1 Release Notes (nvidia.com)

但是pytorch1.7.1官方推荐cuda版本为11.0,对应的cudnn版本为8.0.5.39。如下面的链接:

Previous PyTorch Versions | PyTorch

下面分别是CUDA与CUDNN的官网:

CUDA Toolkit 11.0 Download | NVIDIA Developer

CUDA Deep Neural Network (cuDNN) | NVIDIA Developer

 下载到对应版本后下一步

安装cuda11.0,以管理员身份运行下载的cuda_11

评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值