Ubuntu16.04+CUDA9.0+cuDNN7+yoloV3

一、Ubuntu16.04 GTX 1080Ti+CUDA9.0+GPU driver 390.77+cuDNN7

https://www.jianshu.com/p/a9dcf59bd896

1.1 CUDA与GPU Driver,cuDNN三者版本要匹配

在这里插入图片描述https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

  • 安装CUDA 9.2 cuda-drivers 390.77,显示CUDA driver version is insufficient for CUDA runtime version
  • 重新安装beta版 cuda-drivers 396.18 PASS,但是TensorFlow1.9不支持

1.2 重新安装CUDA9.0和GPU Driver390.77

$ sudo ./cuda_8.0.61_375.26_linux.run --no-opengl-libs

在这里插入图片描述找了很久的原因,为什么Driver Not Selected。其实已经安装成功,可以直接下一步。

sudo nvidia-smi

二、安装 TensorFlow 1.9

https://www.tensorflow.org/install/install_linux

2.1 确定安装哪种 TensorFlow

$ sudo apt-get install cuda-command-line-tools

上述命令已自动实现

$ export LD_LIBRARY_PATH=${LD_LIBRARY_PATH:+${LD_LIBRARY_PATH}:}/usr/local/cuda/extras/CUPTI/lib64

为避免以后系统升级过程中出现 cuDNN 版本冲突,可将 cuDNN 版本保持为 7.0.5

$  sudo apt-mark hold libcudnn7 libcudnn7-dev

日后要允许进行升级,可以取消这一保持行为:

$  sudo apt-mark unhold libcudnn7 libcudnn7-dev

2.2 使用 Virtualenv 进行安装

(tensorflow)$ pip3 install --upgrade tensorflow-gpu # for Python 3.n and GPU

速度太慢,采用pypi 镜像使用帮助
https://mirrors.tuna.tsinghua.edu.cn/help/pypi/

(tensorflow)$ pip3 install https://pypi.tuna.tsinghua.edu.cn/simple some-package

请注意,每次使用 TensorFlow 时,您都必须激活 Virtualenv 环境。如果 Virtualenv 环境当前未处于活动状态,请调用以下命令之一:

$ source ~/targetDirectory/bin/activate      # bash, sh, ksh, or zsh
$ source ~/tensorflow/bin/activate.csh  # csh or tcsh
(tensorflow)$ deactivate 

tensorflow–>targetDirectory

三、安装OpenCV3.4.0

https://blog.csdn.net/qq_36362060/article/details/80739573

pkg-config --modversion opencv #可以查看当前的opencv版本

3.1 OpenCV3.4.2 ./darknet 失败 卸载

https://stackoverflow.com/questions/51002010/opencv-assertion-failed-with-darknet

OPENCV3.4.2
terminate called after throwing an instance of 'cv::Exception'
what():  OpenCV(3.4.2) /home/dir/opencv-3.4.2/modules/core/src/matrix_wrap.cpp:800: error: (-215:Assertion failed) (flags & FIXED_TYPE) != 0 in function 'type'

Darknet won’t compile with opencv 3.4.1. So i suggest downgrading
opencv to 3.4.0.

3.2 安装OpenCV3.4.0

1)安装依赖项

sudo apt-get install build-essential
sudo apt-get install cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev
sudo apt-get install python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev

2)修改opencv编译配置文件
进入到opencv/cmake 目录下,找到OpenCVDetectCUDA.cmake文件并打开,定位到21行,修改set(HAVE_CUDA 1)为set(HAVE_CUDA 0)
3)安装opencv

cd build    #进入到build文件夹
cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..    #开始编译

- 解决ippicv下载问题
https://blog.csdn.net/neilooo/article/details/78425559
- 去github下载对应的库
https://github.com/opencv/opencv_3rdparty/branches/all

ls -a #显示隐藏文件
make -j8   
sudo make install -j8

四、安装yolov3

git clone https://github.com/pjreddie/darknet.git

https://pjreddie.com/darknet/install/#cuda

darknet的MakeFile配置,进入到下载下来的darknet文件夹下,打开Makefile,对如下几个地方进行修改:

CUDA=1
CUDNN=1
OPENCV=1
ARCH= -gencode arch=compute_61,code=[sm_61,compute_61]

图片: https://images-cdn.shimo.im/V5cGG2JIILIklRgs/图片.png

4.1 Compiling With CUDA

GPU=1
./darknet -i 1 imagenet test cfg/alexnet.cfg alexnet.weights
./darknet -nogpu imagenet test cfg/alexnet.cfg alexnet.weights

4.2 Compiling With OpenCV

OPENCV=1
./darknet imtest data/eagle.jpg

make clean
make -j8
wget https://pjreddie.com/media/files/yolov3-tiny.weights  #下载权重文件
./darknet detect cfg/yolov3-tiny.cfg yolov3-tiny.weights data/dog.jpg  #使用下载的权重文件进行图片检测

四、Yolov3-tiny测试结果

在这里插入图片描述
在这里插入图片描述
对自行车和人的识别效果较好,狗的识别率有待提高。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: 要在Ubuntu 16.04上安装CUDA 9.0,必须按照以下步骤进行操作: 1. 首先,从NVIDIA官方网站上下载CUDA 9.0的安装包(通常是一个.run文件)。确保下载的文件和操作系统的位数匹配(32位或64位)。 2. 在终端中导航到下载的安装包所在的目录,并确保安装包有执行权限。若无执行权限,可以使用以下命令添加执行权限: ```bash chmod +x 文件名.run ``` 3. 关闭所有正在运行的图形用户界面。可以通过使用以下命令进行: ```bash sudo service lightdm stop ``` 4. 执行安装脚本以启动安装程序: ```bash sudo ./文件名.run ``` 5. 按照安装程序的提示进行操作。在“Installation Option”选项卡中,选择安装CUDA Toolkit并确保选择与操作系统和硬件兼容的驱动程序版本。 6. 在安装期间,可能会提示你安装NVIDIA驱动程序。如果系统尚未安装适当的驱动程序,则可以选择是以便一同安装驱动程序。 7. 安装完成后,打开终端并将CUDA库和二进制文件的路径添加到环境变量中。可以使用以下命令创建一个新的文件并编辑它: ```bash sudo nano /etc/environment ``` 8. 将以下两行添加到文件的末尾(根据安装路径更改): ``` PATH="/usr/local/cuda-9.0/bin:$PATH" LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH" ``` 9. 保存并退出文本编辑器,并重新启动系统。 10. 系统重启后,可以通过运行以下命令来验证安装是否成功: ```bash nvcc --version ``` 如果安装成功,将会显示CUDA的版本号。 希望以上步骤可以帮助你在Ubuntu 16.04上成功安装CUDA 9.0。 ### 回答2: 首先,您需要确保您的计算机满足CUDA 9.0的系统要求。您需要一张支持CUDA的NVIDIA显卡和安装了Ubuntu 16.04的操作系统。 以下是在Ubuntu 16.04上安装CUDA 9.0的步骤: 1. 首先,下载CUDA 9.0的安装包。可以从NVIDIA的官方网站上下载对应版本的CUDA安装包。 2. 下载完成后,打开终端并切换到CUDA安装包所在的目录。例如,使用以下命令切换到Downloads目录: ``` cd ~/Downloads ``` 3. 接下来,运行以下命令解压CUDA安装包: ``` tar -xvzf cuda_9.0.176_384.81_linux.run ``` 4. 解压完成后,切换到解压后的目录: ``` cd cuda-linux* ``` 5. 运行安装脚本。您可以使用以下命令进行安装: ``` sudo ./cuda-linux*.run ``` 在安装过程中,您需要接受相关许可协议和选择要安装的组件。根据您的需要进行选择并按照提示操作。 6. 安装完成后,您需要配置环境变量。可以使用以下命令打开.bashrc文件: ``` nano ~/.bashrc ``` 在文件末尾添加以下两行代码: ``` export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 保存并关闭文件。 7. 最后,刷新环境变量并确认CUDA安装成功。使用以下命令刷新环境变量: ``` source ~/.bashrc ``` 然后,运行以下命令验证CUDA是否安装成功: ``` nvcc --version ``` 如果安装成功,您将看到CUDA的版本号。 这就是在Ubuntu 16.04上安装CUDA 9.0的步骤。请按照以上步骤进行操作,并确保您满足系统要求。希望对您有所帮助! ### 回答3: 要在Ubuntu 16.04上安装CUDA 9.0,您需要按照以下步骤进行操作: 1. 下载CUDA 9.0安装包:在NVIDIA的官方网站上下载适合您操作系统的CUDA 9.0安装包。 2. 关闭图形界面:在开始安装CUDA之前,您需要先关闭图形界面。按下Ctrl + Alt + F1切换到命令行界面。 3. 卸载旧版本的NVIDIA驱动程序(如果有):如果您之前已经安装了NVIDIA驱动程序,可以通过运行以下命令来卸载它: ``` sudo apt-get remove --purge nvidia* ``` 4. 安装CUDA 9.0:在命令行界面中,导航到您下载的CUDA 9.0安装包所在的目录,并运行以下命令来进行安装: ``` chmod +x cuda_9.0_*.run sudo ./cuda_9.0_*.run ``` 5. 配置环境变量:安装完成后,需要将CUDA的路径添加到系统环境变量中。编辑`~/.bashrc`文件,并在文件末尾添加以下行: ``` export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} ``` 6. 更新环境变量:保存文件后,运行以下命令以更新环境变量: ``` source ~/.bashrc ``` 至此,您已成功在Ubuntu 16.04上安装了CUDA 9.0。您可以通过运行以下命令来验证安装是否成功: ``` nvcc --version ``` 希望这些步骤能帮助您顺利完成CUDA 9.0的安装。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值