YoloV4GPU环境配置

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

你可能感兴趣的:(笔记,tensorflow,深度学习,python)