深度学习框架安装

根据自己的实验需求选择,先下载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

原文地址:https://www.cnblogs.com/ylHe/p/13825365.html