赞
踩
conda create -n <name> python==3.8
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
完整的代码如下:
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
conda install cudatoolkit=11.3.1
conda install cudn=8.2.1
克隆yolov5:
- https://docs.ultralytics.com/tutorials/train-custom-datasets/
- git clone https://github.com/ultralytics/yolov5
cd yolov5
下载依赖包:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple some-package
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。