tensorflow和pytorch安装教程

最新推荐文章于 2024-09-19 10:21:12 发布

浩然harry_M

最新推荐文章于 2024-09-19 10:21:12 发布

阅读量831

点赞数

分类专栏：机器学习文章标签： tensorflow pip 数据库

本文链接：https://blog.csdn.net/weixin_51610638/article/details/120879348

版权

机器学习专栏收录该内容

3 篇文章 0 订阅

订阅专栏

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