最近刚买了新电脑,显卡还不错想跑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

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

被折叠的 条评论
为什么被折叠?



