本次记录在Ubuntu系统上尝试完成Keras版本YOLO V3复现及踩坑超详细过程,期间参考了很多博主的文章,在这一如既往地对他们表示我由衷的感谢。另外,本人小小白一个,文中若有错误还望各位不吝赐教!
【写在前面】
拖了很久,原因是TeamViewer突然连不上服务器了,今天提醒我被商业化了,如果你和我一样不幸,参考帖子:
TeamViewer检测为商用小tips,亲测有效
ok ,开始这次的任务~~~
1 配置环境
1.1 创建环境安装必备软件
Keras版本的Yolo v3的环境是python 3.5.3+Keras 2.1.5+tensorflow 1.6.0
conda create -n py352 python=3.5.2
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple Keras==2.1.5
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow==1.6.0
(小tips:加版本号时,“==”前不要有空格)
1.2 检查电脑配置
主要看符不符合要求~~(这一步熟悉自己电脑的可以略过)~~
一、查看 CUDA 版本:
cat /usr/local/cuda/version.txt
我的是CUDA Version 10.1.168
二、查看GPU型号
lspci | grep -i nvidia
看不懂截个图备用吧先啊哈哈哈哈
三、查看NVIDIA驱动版本
sudo dpkg --list | grep nvidia-*