ubuntu 18.04 安装yolov3 + GPU + CUDA+ cuDNN

最新推荐文章于 2024-08-26 11:40:25 发布

voyagerIII

最新推荐文章于 2024-08-26 11:40:25 发布

阅读量1.2w

点赞数 2

分类专栏：软件安装及环境配置

本文链接：https://blog.csdn.net/debjiu/article/details/80824439

版权

本文详细介绍了在Ubuntu 18.04上安装YOLOv3所需环境的过程，包括CUDA 9.0和cuDNN的安装。首先检查并安装适用于NVIDIA GTX 965M的驱动，然后降级GCC到6.0以下以兼容CUDA。接着，下载并安装CUDA 9.0，确保不安装驱动并调整GCC链接。之后，下载cuDNN并将其内容复制到CUDA安装目录。最后，按照YOLOv3官方指南完成配置。

摘要由CSDN通过智能技术生成

为了测试YOLOv3，折腾了很久.....

机器配置：

GPU: NCIDIA GTX 965M

1.首先是驱动安装：

看一下你的GPU版本和推荐的驱动版本：

$ ubuntu-drivers devices

root@huet-233:/home/huet/pytorch/build# ubuntu-drivers devices

== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001427sv00001462sd0000115Abc03sc02i00
vendor   : NVIDIA Corporation
model    : GM206M [GeForce GTX 965M]
driver   : nvidia-driver-390 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

然后自动安装就可以了：