深度学习框架tensorflow,pytorch,paddlepaddle共同安装的问题

一个深度学习库已经很难装,多个装在一起时很容易出现问题。gpu版的难度大于cpu


这三个一起装一个win10电脑上,先安装Anaconda3-2021.11-Windows-x86_64.exe

(如果前面安装出现错误,最好重新安装)


最好是paddlepaddle先装,再装pytorch,再装tensorflow


第一步,安装paddlepaddle

安装paddlepaddle前,先要确定好cuda的版本,还有cudnn的版本完全一致。即显卡支持,同时paddlepaddle也支持

我是30系统显卡下载的这两个文件

cuda_11.2.0_460.89_win10.exe

cudnn-11.3-windows-x64-v8.2.1.32

用的pip安装

python -m pip install paddlepaddle-gpu==2.2.2.post112 -f https://www.paddlepaddle.org.cn/whl/windows/mkl/avx/stable.html

第二步,安装pytorch

我选择的中cuda 1.11版本,这个版本和上面的11.2不一样,但在30系统显卡下可以运行。

python -m pip install torch==1.8.2+cu111 torchvision==0.9.2+cu111 torchaudio===0.8.2 -f https://download.pytorch.org/whl/lts/1.8/torch_lts.html -i https://pypi.tuna.tsinghua.edu.cn/simple


paddlepaddle,pytorch可以直接安装在本地base中,不用创建虚拟环境

第三步、安装tensorflow

注意不要使用Pip安装,因为pip安装解决不了依赖库cuda,cudnn的问题,还有可能造成其他错误。

下面的命令在新建的conda虚拟环境下运行

conda install -c anaconda tensorflow-gpu


tensorflow不支持Python的高版本,有库冲突,需要创建虚拟环境,如创建python3.7环境,使用conda安装tensorflow,这样不和原来的paddlepaddle,pytorch冲突。否则可能后面是无尽的版本冲突,你装上了这个,另外一个不支持,各种坑。


这样使用paddlepaddle,pytorch可以直接在Anaconda3下直接使用,tensorflow在python3.7环境运行。

编辑于 2022-04-30 11:08