YoloV4GPU环境配置

博主在新电脑上配置YOLOV4的GPU环境,经历了从Tensorflow到Pytorch的过程。首先,配置了tensorflow 2.3的GPU环境,但发现CPU在运行模型,未利用GPU。接着,详细创建了Pytorch的GPU环境,通过两个论坛帖子的帮助,解决了问题。主要挑战在于正确设置Pycharm中的Anaconda虚拟环境和安装GPU支持的Pytorch。总结关键步骤为:1. Pycharm选择Anaconda虚拟环境并添加依赖;2. 安装GPU版Pytorch。

最近刚买了新电脑,显卡还不错想跑YOLO,跑起来发现是CPU在运行,GPU没工作,然后配置了一整天的GPU环境,太折磨人了

一共搭建了两个环境 一个tensorflow:tf2.3。在这个环境里我下了一些需要用到的包。然后又创建了一个torch的gpu环境,这个最重要,下完之后就能用gpu跑了。一共看了两个帖子,都在收藏夹里。

我给这俩帖子直接复制过来。第一个主要是Pycharm里面选择anaconda的虚拟环境,这个一定好配置好。然后配置tensorflow的gpu环境

第二个主要是配置pytorch的gpu环境 第三个结合着看,我用前两个没有讲配置tensorflow的gpu,第三个帖子里有

总结来说 我就是pytorch的GPU没下好,搞了一整天。。。

也就是两点1.pycharm里面选择好创建好的环境,在环境里面加入相应的依赖2.创建pytorch的gpu环境

如何在pycharm中配置anaconda的虚拟环境_肆十二的博客-CSDN博客_pycharm使用anaconda虚拟环境

使用anaconda配置gpu版本的torch==1.7.1(非30系列以上显卡也可用)_大气层煮月亮的博客-CSDN博客_anaconda调用gpu

2021年Windows下安装GPU版本的Tensorflow和Pytorch_肆十二的博客-CSDN博客

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值