Ubuntu配pytorch版yolov5环境

链接：https://pan.baidu.com/s/19kJmuSuSWB3B3gHUZ2oKbg 提取码：r7wp*
#此链接包含Anaconda安装包、pycharm安装包、torch-1.8.0、torchvision-0.9.0以及yolov5-v5.0源码
#配环境前可以先把这五个拷贝到home目录下

#注意网盘里的torch和torchvision安装是cuda111版本的（cuda11.0、11.2好像也可以用），若自己电脑是其他cuda版本，可以从该网站下载对应安装包
https://download.pytorch.org/whl

一、安装Anaconda

bash Anaconda3-2020.11-Linux-x86-64.sh
please answer 'yes' or 'no'       
键入 yes
Anaconda3 will now be installed into this location:
/home/用户名/anaconda3       
键入 回车
Do you wish the installer to initialize Anaconda3
by running conda init? [yes|no]
键入 yes

二、安装使用pycharm

tar -xvzf pycharm-community-2021.2.tar.gz
cd pycharm-community-2021.2/bin
sh pycharm.sh

三、创建虚拟环境

1、添加清华源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
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/r
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
conda config --set show_channel_urls yes

2、新建虚拟环境

conda create -n yolov5 python=3.8

3、进入虚拟环境

conda activate yolov5

4、安装pytorch

pip install  （把torch安装包拖进终端）
pip install  （把torchvision安装包拖进终端）

##也可从pytorch官网安装，根据所需torch版本和电脑cuda版本，拷贝官网提供的指令即可
5、测试
#打开终端

python
import
torch.cuda.is_available()  #输出true即安装成功
print(torch.__version__)  #查看torch版本

四、拷贝yolov5源码

git clone https://github.com/ultralytics/yolov5.git

#此处拷贝的是最新版yolov5代码，若需使用之前版本的代码，操作如

git clone -b v5.0 https://github.com/ultralytics/yolov5.git

#也可直接解压从网盘下载的yolov5-v5.0源码压缩包

unzip yolov5-v5.0.zip

五、安装yolov5源码所需依赖库

cd yolov5代码目录
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

六、测试

$ python detect.py --source "data/images/bus.jpg" --weights="weights/yolov5s.pt"