tensorflow和pytorch安装教程

tensorflow安装:

1、CPU版本tensorflow安装:
直接在anaconda prompt里面输入命令:pip install tensorflow 这个命令安装的是最新版本

2、GPU版本tensorflow安装:
在anaconda prompt里面输入命令:
pip install cudatoolkit=版本号
pip install cudnn=版本号
pip install tensorflow-gpu=版本号

要注意的是,cuda、cudnn、tensorflow版本号要对应。

在这里插入图片描述

前两行命令是安装cuda和cudnn。至于cuda和cudnn这两个东西是什么,简单解释一下,计算机gpu计算,是把数据放到gpu上,cuda相当于一个连接的东西。cudnn则是专门用来完成深度学习的工具。(这个解释很简单笼统,具体解释在文章末尾)

pytorch 安装

我们直接进入pytorch的官网,官网提供GPU以及CPU版本的安装命令。
找到如下界面:
在这里插入图片描述
如图所示,选择你对应的系统、语言以及CUP或GPU,图中CUDA10.2 CUDA11.1便是对应的GPU版本。然后将下面给的命令输入anaconda中,就可以了。就是这么简单。

CUDA和cuDNN

在这里,顺便解释一下什么是CUDA和cuDNN

CUDA与cuDNN的关系:
CUDA看作是一个工作台,上面配有很多工具,如锤子、螺丝刀等。cuDNN是基于CUDA的深度学习GPU加速库,有了它才能在GPU上完成深度学习的计算。它就相当于工作的工具,比如它就是个扳手。但是CUDA这个工作台买来的时候,并没有送扳手。想要在CUDA上运行深度神经网络,就要安装cuDNN,就像你想要拧个螺帽就要把扳手买回来。这样才能使GPU进行深度神经网络的工作,工作速度相较CPU快很多。
这段解释引自:https://www.jianshu.com/p/622f47f94784

关于CUDN更详细的解释,可以参考一下这篇博文:https://blog.csdn.net/xiaoxiaolibai/article/details/104770046

猜你喜欢

转载自blog.csdn.net/weixin_51610638/article/details/120879348