根据自己的实验需求选择,先下载whl文件,再pip install。(直接pip 下载速度实在是太慢了。。。)
1. pytorch-gpu安装
https://download.pytorch.org/whl/cu90/torch_stable.html
我选择的是:torch-1.0.0-cp36-cp36m-win_amd64.whl
如果网速可以,也可以在https://pytorch.org/get-started/previous-versions/
2.tensorflow-gpu安装
https://pypi.org/project/tensorflow-gpu/#files
我选择的是:tensorflow_gpu-2.3.1-cp36-cp36m-manylinux2010_x86_64.whl
pip install后,还两个小错误,安装提示缺啥pip啥就行。
3.keras 安装
https://pypi.org/project/Keras/#files
我选择的是:Keras-2.4.3-py2.py3-none-any.whl
4.opencv 安装
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple