pytorch环境配置教程(基于Anaconda)

pytorch环境配置教程(基于Anaconda)

1、下载Anaconda
2、创建虚拟环境conda create -n yolov5 python=3.7
3、conda activate yolov5
4、换源
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/msys2/
conda config --add channels https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/
conda config --set show_channel_urls yes
5、pytorch版本选择以及安装
(1)输入conda search pytorch找到pytorch版本号,要求1.9.0以上,以及cuda11.0以上,这里我们选择1.9.1以及cuda11.1
pytorch环境配置教程(基于Anaconda)_第1张图片
(2)安装所选版本号conda install pytorch=1.9.1=py3.7_cuda11.1_cudnn8.0.5_0
(3)找torchvision版本:conda search torchvision
找到torchvision是0.10.1,py是37,cuda是111的torchvision版本
(4)找到torchvision版本后conda install torchvision=0.10.1=py37cuda111hfcef1eb_0_cuda
(5)查看我们的torchvision版本conda list torchvision
(6)根据教程下载Yolov5: https://github.com/ultralytics/yolov5

首先,执行
git clone https://github.com/ultralytics/yolov5
如果Git没下载,就用https://blog.csdn.net/Andone_hsx/article/details/87937329
再执行
pip install -r requirements.txt -i http://pypi.douban.com/simple --trusted-host pypi.douban.com

你可能感兴趣的:(yolov5实战,pytorch,python,深度学习)