机器学习常用python包

 

(py37) ai@ai:~$ pip freeze |grep -v '@'
astor==0.8.1
certifi==2021.5.30
chardet==4.0.0
cycler==0.10.0
gast==0.2.2
google-pasta==0.2.0
h5py==2.8.0
idna==2.10
jieba==0.42.1
joblib==1.0.1
mkl-fft==1.3.0
mkl-service==2.3.0
olefile==0.46
opencv-python==4.5.2.54
pandas==1.2.4
protobuf==3.14.0
py4j==0.10.4
pydot==1.3.0
pydotplus==2.0.2
pyplotz==0.24
pyspark==2.2.1
requests==2.25.1
scikit-learn==0.24.2
sklearn==0.0
tensorboard==1.15.0
tensorflow==1.15.0
tensorflow-estimator==1.15.1
termcolor==1.1.0
threadpoolctl==2.1.0
torch==1.8.1
torchaudio==0.8.0a0+e4e171a
torchvision==0.2.1
tqdm==4.61.1
urllib3==1.26.5
webencodings==0.5.1
Werkzeug==0.16.1
wrapt==1.12.1
xgboost==1.4.2

 

去掉一部分之后剩下的包如下：

jieba==0.42.1
opencv-python==4.5.2.54
pandas==1.2.4
pydot==1.3.0
pydotplus==2.0.2
pyplotz==0.24
pyspark==2.2.1
requests==2.25.1
scikit-learn==0.24.2
tensorboard==1.15.0
termcolor==1.1.0
threadpoolctl==2.1.0
torch==1.8.1
torchaudio==0.8.0a0+e4e171a
torchvision==0.2.1
tqdm==4.61.1
xgboost==1.4.2

 

其中的两个重量级模块: pytorch与tensorflow

tensorflow在python3.7中可安装1.15.10, 在python3.8中可安装2.5.0，如果是python2.7环境可能要降至1.1.0

pytorch兼容性比较好，不需要注意与python版本的配合，通常安装最新版即可

另外，tensorflow最好与pytorch分环境安装，不要安装在一个环境中，避免相互影响

 

pytorch推荐安装方式

去官网https://pytorch.org/get-started/locally/

根据自己的情况进行安装

https://pytorch.org/get-started/locally/

 

pip3 install torch torchvision torchaudio

这三个一块安装时，torch1.9版本，单安装torch时，目前默认安装的是1.8版本，然后就安装不上torchvision了
可以先卸载pytorch，再三个一起安装；要么三个一起指定版本，要么都不指定让pip自动权衡
pip3 uninstall torch

 

pytorch 优点
- 兼容性好，通常安装最新版即可
- pytorch 与tensorflow都可以求导
- pip install tensorflow==1.1.0
- tensorflow推荐安装1.1.0版本

 

整理到文件中进行安装

pip install -r ml.txt

 

快速安装（最实用）

pip install -r requirements.txt -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com