Windows下Anaconda中配置Yolov5环境
Anaconda安装pytorch
1、换源
增加下载速度,换源到国内镜像源,在cmd控制台或者powershell下输入命令即可
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 --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/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 --set show_channel_urls yes
2、创建虚拟环境
打开Anaconda Prompt,此时应该是在(base)环境下,直接输入
conda create -n pytorch python=3.7
创建了一个名为pytorch的虚拟环境,环境中python版本为3.7
以后再打开Anaconda Prompt, 命令行输入
conda activate pytorch
可直接进入虚拟环境。
3、安装pytorch
直接进入pytorch官网,附上连接:pytorch官网地址
官网页面如下,需要注意:30系列显卡需要装CUDA11以上的版本!!
由于之前换了源,直接在命令行输入一下命令,不用-c pytorch ,这个是从官方地址下载,国外会比较慢。
conda install pytorch torchvision torchaudio cudatoolkit=11.3
安装时间会稍微长一些,不用操作等一会儿就安装好了。
4、下载Yolov5
附上官方地址,我下的是Yolov5-v5.0版本的
https://github.com/ultralytics/yolov5
Github主页,可以自选版本。
按照项目文件里的requirement.txt安装所需所需包。
pip install -r requirement.txt
pycocotools包会出现问题。
找下面这位兄弟。他已经共享了。
https://blog.csdn.net/oijdkd/article/details/120344665
欢迎大家点赞收藏,后面有空再介绍利用TensorRt在Jetson嵌入式系列板子上的部署。