在跑Github某个项目的时候,发现环境配置要求
Python >= 3.7
Pytorch == 1.5.x
目前官网上安装,默认是1.7.0版本的
安装Pytorch1.5.1版本指令
pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple
采用豆瓣源安装
速度还是挺快的
安装完成!
查看一下是否支持CUDA和Pytorch版本号
接下来学习yolov5可能会用到