TensorFlow
pip install tensorflow == x.x.x -i 换源
注意版本问题,Python版本和Tensorflow版本之间的支持性。比如Python3.7以及更高版本在之前没有TensorFlow版本支持。
Keras
Keras依赖TensorFlow环境。注意版本问题,默认是最新版本,但是最的Keras需要高版本的Tensorflow支持。如果TensorFlow版本不支持,建议安装更低版本的Keras,不建议升级TensorFlow,因为最高版本的TF相关资源较少,工业界更多使用TF1.4。
pip install keras == x.x.x #安装包小,不需要换源
Pytorch
去官网选择合适的pip指令,注意GUP驱动的问题。
pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
换源清华源和豆瓣源都不太好使,建议pip的时候科学上网。
成功安装 python3.7.6 + tensorflow1.15.5 + keras2.3.1 + pytorch1.7.1
后记:Pytorch1.7会报错,缺少C++库支持,建议安装1.6并按照提示的链接安装C++的库。