Anaconda3 + Pytorch1.8 + Cuda10.2 + cuDnn8.2.0 + opencv3.2.0
opencv3.2.0
之前装orb-slam2的时候装过opencv3.2.0
pkg-config --modversion opencv
安装CUDA10.2
之前装zed2摄像头时已经安装过cuda10.2
nvcc -V
安装cuDNN8.2.0
解压cuDNN压缩文件
sudo tar -xzvf cudnn-10.2-linux-x64-v8.2.0.53.tgz
会在当前文件夹下生成一个名为cuda的文件夹
然后输入以下命令:
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.2/lib64/
sudo cp cuda/include/cudnn* /usr/local/cuda-10.2/include/
sudo chmod a+r /usr/local/cuda-10.2/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.2/lib64/libcudnn*
到这里cuDNN就安装完成了
Anaconda3
下载
wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh
安装
bash Anaconda3-2021.05-Linux-x86_64.sh
安装成功
编辑~/.bashrc文件,加入
export PATH=~/anaconda3/bin:$PATH
更新环境变量
source ~/.bashrc
输入conda list
添加Anaconda国内镜像配置
清华提供了Anaconda仓库的镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
安装Pytorch1.8
注意:如果使用yolov5版本v5.0以上的代码,使用pytorch1.8
基础语法:
conda create -n yolov5_test python==3.8
conda activate yolov5_test
conda remove -n yolov5_test --all
conda deactivate yolov5_test
开始安装
首先为pytorch创建一个anaconda虚拟环境,环境名字可自己确定,这里使用yolov5_test作为环境名:
conda create -n yolov5_test python==3.8
安装成功后激活yolov5_test环境
conda activate yolov5_test
问题:
运行:
source activate
重新激活
在所创建的环境下安装pytorch的1.8版本
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
注意:10.2处应为cuda的安装版本号
编辑~./bashrc文件,设置使用yolov5_test环境下的python3.8
alias python='~/anaconda3/envs/yolov5_test/bin/python3.8'
更新环境变量
source ~/.bashrc
该命令将自动回到base环境,再执行
conda activate yolov5_test
yolov5项目克隆安装
下载文件命名为yolov5_test
git clone https://github.com/ultralytics/yolov5.git yolov5_test
安装所需库
在yolov5_test目录下执行
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
下载权重文件 https://github.com/ultralytics/yolov5/releases/ ,放置在weights文件夹下
测试
python3 detect.py --source ./data/images/ --weights weights/yolov5s.pt
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)