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