要运行yolov5,首先要安装pycharm,pycharm随便安装一个,没有版本限制。还要安装一个Anaconda,这个版本需要注意。可以参考别人的博客
(9条消息) Anaconda版本与Python版本对应关系_小徐勿忘初心的博客-CSDN博客_python3.7对应的anaconda版本
例如,我要在Anaconda里面创建python==3.7版本的环境,我要安装Anaconda 2018.12,这个版本对应不是一一对应的,而是多对多。
安装Anaconda后,打开Anaconda Prompt更改conda源(后续安装第三方库可以加快速度)
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
检查是否改好:
conda config --show channels
然后创建虚拟环境
conda create -n pytorch python=3.7
规则是:conda create -n 自己起的虚拟环境名 python==版本号,我这里是3.7
进入到虚拟环境中
conda activate pytorch
然后查看自己nvidia的CUDA版本号
nvidia-smi
我的CUDA版本号是11.2,在网上找到的安装对应的pytorch。
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.2 -c pytorch -c conda-forge
然后下载yolov5的源代码,放到pycharm里,在terminal里输入
pip install -r requirements.txt
提示:电脑里要有visual studio,下载requirements.txt里面的东西需要一些插件的支持。
这是需要安装的东西:不要在pycharm下安装,速度太慢了。
pip install Cython
pip install numpy
pip install opencv-python
pip install matplotlib
pip install pillow
pip install tensorboard
pip install PyYAML
pip install torchvision
pip install scipy
pip install tqdm
有些东西安装需要特定的版本:
torchvison和python、pytorch和cuda的版本要对应。
pip install opencv-python也要看版本
有些东西可以下载下来再安装
pip install 具体路径。
最后,哪里出现问题了就复制下来多百度。大家都是面向浏览器编程
ModuleNotFoundError: No module named 'torch'
1.查看你现在用的python版本是不是你安装torch的版本 ;
2、看一下cuda版本与torch'版本是否对应;