利用Anaconda安装pytorch和paddle深度学习环境+pycharm安装)
大体步骤参考此链接https://blog.csdn.net/didiaopao/article/details/119787139#comments_28749195,安装过程当中存在以下问题:
跑yoloV5框架不需要安装paddle包(可以忽略安装paddle包这一步骤)。由于yoloV5框架的问题,所以本次安装与链接中的教学保持高度一致,其余需注意以下几点:需要重新安装cuda版本与cuda对应的pytorch,本次安装cuda采用10.2版本,pytorch版本也不适宜太低,参考yoloV5官网推荐版本
torch版本大于等于1.7,创建虚拟环境时python版本创3.8
所以本次安装安装1.7.1版本,在pytorch官网找到1.7.1版本用 pip install 指令下载
由于CUDA版本10.2那个下载会报错,所以后来采用pip install torch1.7.1+cu101 torchvision0.8.2+cu101 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
指令进行下载。(-i https://pypi.tuna.tsinghua.edu.cn/simple)这一串是清华源,可以换其他源,另外国内的源下载时不能挂梯子,不然访问不了国内源会报错