GTX1050配置YOLOv5深度学习win10环境 |
您所在的位置:网站首页 › 1050显卡能用win10吗 › 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 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |