Python 深度学习常用包汇总

Python 深度学习常用包汇总

比较常用Pytorch框架，记录一下常用的包，方便重建深度学习环境。如果没有特别提示，均是在Windows下试验的。

更新历史

2021/2/28
2021/3/6
2021/5/22
2022/4/22
2023/8/18
2023/9/25 增加飞浆paddlepaddle

1、框架

不同框架一般安装在不同环境中
最新Anaconda 创建Python3.6，Python3.7虚拟环境，为Pytorch,tensorflow创建单独环境
检验 pytorch,tensorflow，paddle，mxnet 深度学习框架是否正确支持GPU功能
python给pip修改镜像-cmd命令修改版
在anaconda环境中使用conda命令安装cuda、cudnn、tensorflow（-gpu）、pytorch

1.1 pytorch

官网首页：https://pytorch.org/

旧版安装命令地址: Previous PyTorch Versions | PyTorch

当前安装版本：1.7.1 gpu 版本

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
1

pytorch 1.10.1 gpu 版本

pip install torch==1.10.1+cu111 torchvision==0.11.2+cu111 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html
1

30系显卡推荐安装torch1.10.0 以及cu111以上版本

1.2 飞浆paddlepaddle

官网 飞浆paddlepaddle: https://www.paddlepaddle.org.cn/install

版本2.4.2

# Linux CUDA 11.2
python3 -m pip install paddlepaddle-gpu==2.4.2.post112 -f https://www.paddlepaddle.org.cn/whl/linux/mkl/avx/stable.html


# Windows CUDA 11.2
python -m pip install paddlepaddle-gpu==2.4.2.post112 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html
1
2
3
4
5
6

如果项目是旧代码，推荐使用2.4.2以下版本，因为我遇到新版本2.5.x有些api被更改了。

1.3 tensorflow 1.15 cpu ，gpu版

官网：https://tensorflow.google.cn/install

python版本要是3.5 - 3.7, 推荐使用python3.7，因为3.6及以下版本在vscode不支持debug。

pip install tensorflow==1.15
1

# gpu版
pip install tensorflow-gpu==1.15
1
2

linux系统最好的安装方法

conda install -c conda-forge tensorflow-gpu==1.15.0
1

这样会顺便安装cudatoolkit和 cudnn插件不用全局安装GPU的cuda了，因为配置全局cuda实在有的难

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/641336