搭建YOLOv3环境
机器硬件配置
GPU : RTX2070 8G
内存:16G
硬盘:500G机械
主板: 微星H110M PRO-A
机器系统
CentOS 7.6
移步至此链接在CentOS 7下搭建机器学习环境
安装完显卡在BIOS
上找到Integrated Devices
在Embedded Video Controller
选项, 将这个设置为Disabled
禁用集成显卡
根据自己的显卡安装驱动
安装检查显卡程序
$ sudo yum install nvidia-detect
$ nvidia-detect -v
Probing for supported NVIDIA devices...
[10de:1f07] NVIDIA Corporation TU106 [GeForce RTX 2070 Rev. A]
This device requires the current 430.34 NVIDIA driver kmod-nvidia
[8086:5912] Intel Corporation HD Graphics 630
根据 nvidia-detect 的输出信息,可以知道显卡的型号,以及要使用的驱动版本 430.34 。
$ yum -y install kmod-nvidia
$ nvidia-smi
Wed Aug 7 23:14:08 2019
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 430.34 Driver Version: 430.34 CUDA Version: 10.1 |
|-------------------------------+----------------------+----------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
|===============================+======================+======================|
| 0 GeForce RTX 2070 Off | 00000000:01:00.0 Off | N/A |
| 41% 26C P8 9W / 185W | 4657MiB / 7982MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
+-----------------------------------------------------------------------------+
| Processes: GPU Memory |
| GPU PID Type Process name Usage |
|=============================================================================|
| No running processes found |
+-----------------------------------------------------------------------------+
cuda安装官方教程
参考资源1
参考资源2
参考资源3
增加用户
安装Cuda错误信息的相应解决办法1
安装Cuda错误信息的相应解决办法2