GTX1050配置YOLOv5深度学习win10环境

YOLOv5要求:Python》=3.7 pytorch》=1.7

1.驱动下载

下载旧版本英伟达驱动

关于何如在英伟达官网上下载历史驱动的方法_BeatifulBoyBoy的博客-CSDN博客_英伟达历史驱动

我选择的是419.17版本

2.CUDA及cudnn 安装

根据驱动版本(419.17)找到控制面板,找到组件查看cuda版本(我的是10.1),所以选择安装10.1版本。

点击进入CUDA Toolkit Archive网址,下载其中的cuda10.1版本。

cudnn根据cuda安装对应的版本,将解压的三个文件复制到cuda安装的文件中。

深度学习环境搭建(GPU)CUDA安装(完全版)_小邢同学的博客-CSDN博客_cuda搭建

3.anaconda安装

下载Anaconda3-2020.02-Windows-x86_64

选择自动安装环境变量

配置清华镜像源

 Anaconda配置清华镜像源_克豪的博客-CSDN博客_anaconda 清华镜像

4.安装pytorch

首先配置虚拟环境

# 创建
conda create -n env_name python=x.x

我选择的Python3.7

然后进入pytorch官网。点击进入Pytorch官网

选择之前的版本

安装pytorch1.7.1,cuda选择对应的10.1版本

 

在自己创建的虚拟环境中输入即可安装成功。

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

猜你喜欢

转载自blog.csdn.net/weixin_52950958/article/details/125402164