当前位置: 首页 > news >正文

【目标检测】Anaconda+PyTorch配置

前言

  本文主要介绍在windows系统上的Anaconda、PyTorch关键步骤安装,为使用yolo所需的环境配置完善。同时也算是记录下我的配置流程,为以后用到的时候能笔记查阅。

Anaconda

软件安装

Anaconda官网:https://www.anaconda.com/

另外,Anaconda下携带的conda的基本命令,建议可以查看菜鸟教程的介绍。Anaconda 教程

  在上数官网完成安装后,进入系统自带命令行Ctrl+R输入cmd或者 Anaconda携带的Anaconda Prompt (Anaconda)都可以,二选一即可,我通常选用的是系统自带的命令行。

可以尝试看看Anaconda有没有被安装

conda -V

在这里插入图片描述

创建环境

继续接着在命令行里操作。以下步骤仅供参考具体,按照个人配置。

  1. 创建环境,指定环境名词,以及python版本

    conda create -n pytorch python=3.8
    

    在这里插入图片描述

  2. 进入环境

    conda activate pytorch
    

    在这里插入图片描述

    虚拟环境,共有两个包管理,是可以同时用的,分别是conda,pip。

PyTorch

  PyTorch是开源的Python机器学习库,在刚才创建好的环境中下载。并且因为我的电脑是有GPU的,所以下载流程按照安装GPU版本的走。假如是要安装CPU版本的话,只要相关包能下载就行,用conda或pip关系不大。安装GPU的话,我这边是选择用pip。

下述命令的执行都是要确保在虚拟环境中执行,也就是上述的(pytorch) C:\Users\XYZ>

当然可以先到PyTorch官网,按自身环境在选择器中选择,就能给出相应下载命令。[PyTorch下载选择器](conda install pytorch torchvision torchaudio cpuonly -c pytorch)

在这里插入图片描述

CPU版本

换源命令,逐行执行:

# 添加清华镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --set show_channel_urls yes

下载命令,纯CPU版本

conda install pytorch torchvision torchaudio cpuonly

GPU版本

  尝试过用conda安装,试过换源等还是CPU版本的,网上又说是什么没有对应的GPU版本之类。所以最后选择用pip安装。

  1. 首先在命令行中,查看CUDA最高支持版本

    在这里插入图片描述

    最高支持的CUDA版本为12.2

  2. 我看到在PyTorch下载选择器上有,CUDA12.1版本的相关,下载的选项,就打算下载那个。但当时电脑版本的CUDA版本不匹配,就要去英伟达官网下载所需的版本。(非必要,假如版本已经对应)

    [英伟达-CUDA历史版本](CUDA Toolkit Archive | NVIDIA Developer)

    可以下述命令查看当前电脑的CUDA版本

    nvcc -V
    

    在这里插入图片描述

  3. 下载PyTorch,安装选择选择器命令下载。

    在这里插入图片描述

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
    

    不过要是按照上述命令下载的话,因为是国外源下载极慢,而且文件是2G左右,要是小一点还能接收。我记得网上有相对应的办法,可以从下述网址:download.pytorch.org/whl/torch_stable.html选定相对应的torch,torchvision的GPU版本文件,手动下载,在基于pip install 包名(该包一定要在目前命令行下,才能找到)导入。不过我嫌对应规则太绕了,就没看


      我的方法是,起始大的特殊的只有torch这个GPU版本的文件,我先是调用(在创建好的虚拟环境中)pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121,它会给出要下载包的名称,Ctrl+C中止下载后,在复制名称到download.pytorch.org/whl/torch_stable.html网址搜索下载,在导入,在重新执行上述pip3 inst...(省略)命令,重新拉取下载,其它的包体积都较小,慢点都没事了,或者加个国内源都可以。

  4. 检查 GPU 驱动程序和 CUDA 是否已启用

      检查 GPU 驱动程序和 CUDA 是否已启用并由 PyTorch 访问,请运行以下命令以返回是否启用了 CUDA 驱动程序:(相当于能启用GPU)

    python 
    import torch 
    torch.cuda.is_available()
    

    在这里插入图片描述

后续

这里,提下后续我在Pycharm上运行YOLO-V5的detect.py,出了问题:

在这里插入图片描述

后来,发现还torchvision没有按照pip命令,下载和torch对应的版本,可能还是CPU版本,就一直运行不起来。解决方法:回到Anaconda创建的虚拟环境中,卸载掉torchvision,还是去download.pytorch.org/whl/torch_stable.html调选对应的版本。

例如我上文中共在网址下载过这两个:

torch-2.3.1+cu121-cp38-cp38-win_amd64.whl

torchvision-0.18.1+cu121-cp38-cp38-win_amd64.whl

反正这里挺疑惑,我当时也正好截了图,明明下载对了,怎么后面又变回正常版本呢

在这里插入图片描述

相关文章:

  • 北京网站建设多少钱?
  • 辽宁网页制作哪家好_网站建设
  • 高端品牌网站建设_汉中网站制作
  • 图像处理 -- ISP调优(tuning)的步骤整理
  • 2024 HNCTF PWN(hide_flag Rand_file_dockerfile Appetizers TTOCrv_)
  • 以Zookeeper为例 浅谈脑裂与奇数节点问题
  • 东京裸机云多IP服务器全面分析
  • 数学建模学习(2)——决策树
  • OpenCV 安装与基础使用教程(Python)
  • RabbitMQ的学习和模拟实现|GTest测试框架的介绍和简单使用
  • 数据结构代码
  • Git基本使用
  • 3D建模软件--犀牛Rhino for Mac
  • Python应用—浅谈利用opencv去除水印
  • 创建最佳实践创建 XML 站点地图--SEO
  • 谷粒商城实战笔记-42-前端基础-Vue-生命周期和钩子函数
  • 深入浅出WebRTC—ULPFEC
  • 挖掘基于边缘无线协同感知的低功耗物联网 (LPIOT) 的巨大潜力
  • [微信小程序] 使用ES6特性Class后出现编译异常
  • 【391天】每日项目总结系列128(2018.03.03)
  • 4月23日世界读书日 网络营销论坛推荐《正在爆发的营销革命》
  • Angular 响应式表单 基础例子
  • iOS 系统授权开发
  • JavaScript函数式编程(一)
  • Java面向对象及其三大特征
  • Map集合、散列表、红黑树介绍
  • SegmentFault 技术周刊 Vol.27 - Git 学习宝典:程序员走江湖必备
  • Storybook 5.0正式发布:有史以来变化最大的版本\n
  • 技术攻略】php设计模式(一):简介及创建型模式
  • 前端设计模式
  • 使用权重正则化较少模型过拟合
  • 一道闭包题引发的思考
  • 一道面试题引发的“血案”
  • 新年再起“裁员潮”,“钢铁侠”马斯克要一举裁掉SpaceX 600余名员工 ...
  • ​​​​​​​ubuntu16.04 fastreid训练过程
  • ‌前端列表展示1000条大量数据时,后端通常需要进行一定的处理。‌
  • #微信小程序(布局、渲染层基础知识)
  • #我与虚拟机的故事#连载20:周志明虚拟机第 3 版:到底值不值得买?
  • (附源码)node.js知识分享网站 毕业设计 202038
  • (附源码)计算机毕业设计ssm-Java网名推荐系统
  • (四)opengl函数加载和错误处理
  • (一)RocketMQ初步认识
  • (原創) 博客園正式支援VHDL語法著色功能 (SOC) (VHDL)
  • .NET 6 在已知拓扑路径的情况下使用 Dijkstra,A*算法搜索最短路径
  • .net core webapi 部署iis_一键部署VS插件:让.NET开发者更幸福
  • .net core 的缓存方案
  • .net core 外观者设计模式 实现,多种支付选择
  • .Net MVC + EF搭建学生管理系统
  • .net 连接达梦数据库开发环境部署
  • .net 前台table如何加一列下拉框_如何用Word编辑参考文献
  • .NET 药厂业务系统 CPU爆高分析
  • .NET平台开源项目速览(15)文档数据库RavenDB-介绍与初体验
  • .net通用权限框架B/S (三)--MODEL层(2)
  • .NET未来路在何方?
  • :中兴通讯为何成功
  • [ACL2022] Text Smoothing: 一种在文本分类任务上的数据增强方法
  • [BUAA软工]第一次博客作业---阅读《构建之法》
  • [c#基础]值类型和引用类型的Equals,==的区别