安装pytorch之前,需要安装cuda和cudnn,具体参考教程:
https://blog.csdn.net/yldmkx/article/details/107139863
建议使用cuda 10.2 版本;
安装好之后,在anaconda环境里面新建pytorch-env的虚拟环境,
然后手动安装pytorch,因为直接在anaconda里面搜索pytorch进行安装,可能会导致
torch.cuda.is_available() 结果是false;
先手动切换到pytorch-env虚拟环境,然后输入conda指令进行安装,具体安装指令可以在pytorch官网查询,
https://pytorch.org/get-started/previous-versions/,pytorch使用手动安装方法,和cuda版本对应;
这样的话,我们可以保证我们按照的是pytorch gpu版本以及和cuda版本号进行对应起来;
如果经常用到pip install -r requirements.txt,
可以在anaconda里面新建虚拟环境,然后使用git工具,
先conda env list,
然后source activate yolov3-pytorch,
最后执行pip install -r requirements.txt即可,
如果有安装失败的包,建议先在 requirements.txt注释掉,然后手动通过pip 或者 conda 命令进行安装。