栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 系统运维 > 运维 > Linux

Ubuntu18.04配置yolov5环境

Linux 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

Ubuntu18.04配置yolov5环境

Python3.7+nvidia-470+cuda11.4+cuDNN+Anaconda3++pytorch1.11
搞了三天,问题一是pytorch1.11,安装要用国内源,二是不熟悉conda环境

安装Python3.7

安装Python3.7
下载网址:https://www.python.org/downloads/source/
sudo cp -r /当前目录/Python-3.7.0 /usr/local/python
cd /usr/local/python/Python-3.7.0
sudo ./configure
sudo make
sudo make install
问题:ModuleNotFoundError: No module named ‘_ctypes
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install build-essential python-dev python-setuptools python-pip python-smbus build-essential libncursesw5-dev libgdbm-dev libc6-dev zlib1g-dev libsqlite3-dev tk-dev libssl-dev openssl libffi-dev

安装显卡驱动nvidia-470

ubuntu-drivers devices
系统推荐:driver : nvidia-driver-470 - distro non-free recommended
点击附加驱动对应的驱动左侧,应用更改。

reboot
重启时进入 perform mok management 界面,选择第二个 enroll mok,选择continue,再选择yes,输入之前的 secure boot 密码。
nvidia-smi//用于支持driver API的必要文件(如libcuda.so)是由GPU driver installer安装的。
nvcc -V//用于支持runtime API的必要文件(如libcudart.so以及nvcc)是由CUDA Toolkit installer安装的。

安装cuda11.4

CUDA Toolkit 11.4 Update 4 Downloads
官方:wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin
sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/11.4.4/local_installers/cuda-repo-ubuntu1804-11-4-local_11.4.4-470.82.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1804-11-4-local_11.4.4-470.82.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu1804-11-4-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

配置cuda环境

sudo gedit /etc/profile
文末添加:

export CUDA_HOME=/usr/local/cuda
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

source /etc/profile
sudo gedit ~/.bashrc
文末添加:

### cuda path
export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH 

source ~/.bashrc
sudo gedit /etc/ld.so.conf
文末添加:

/usr/local/cuda/lib64

sudo ldconfig
nvcc -V
此时和nvidia-smi的cuda一致。

安装cuDNN

Download cuDNN v8.2.4 (September 2nd, 2021), for CUDA 11.4
sudo cp cuda/lib64/* /usr/local/cuda/lib64
sudo cp cuda/include/* /usr/local/cuda/include
cat /usr/local/cuda/include/cudnn_version.h

安装anaconda

步骤没问题,要选好自己的版本,我的后来提示更新,已换conda 4.12.0
查看版本:Packages included in Anaconda 2021.05 for 64-bit Linux with Python 3.6
下载
Anaconda3-2021.05-Linux-x86_64.sh
x86 是32位,x86_64 和 x64 以及AMD64 是64位
lanc@lanc-Y7000P:~/下载$ bash ‘/home/lanc/下载/Anaconda3-2021.05-Linux-x86_64.sh’
记录安装位置:/home/lanc/anaconda3
lanc@lanc-Y7000P:~$ conda --version
版本:conda 4.10.1
Ubuntu打开终端时自动退出 base 虚拟环境命令
进入:conda activate

安装pytorch

conda activate
conda create -n pytorch python=3.7
conda activate pytorch
查看版本
添加源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/menpo/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

安装:
conda install pytorch torchvision torchaudio cudatoolkit=11.3

安装yolov5

下载yolov5
conda activate
conda create -n yolov5 python=3.7
conda activate yolov5
pip3 install -r requirements.txt
python3 detect.py

python detect.py --source 0  # 笔记本相机
                            file.jpg  # image 
                            file.mp4  # video
                            path/  # directory
                            path/*.jpg  # glob
                            'https://youtu.be/NUsoVlDFqZg'  # YouTube video
                            'rtsp://example.com/media.mp4'  # RTSP, RTMP, HTTP stream

python3 detect.py --source ./data/images/ --weights ./weights/yolov5m.pt --conf 0.5
(置信度0.5:–conf 0.5)

ERROR: Command “python setup.py egg_info” failed with error code 1 in /tmp/pip-build-tkzbit6g/opencv-python/
解决方式
ERROR: pip’s dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
launchpadlib 1.10.6 requires testresources, which is not installed.
解决方式

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/826817.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号