安装tensorflow-gpu

最新推荐文章于 2022-07-09 18:08:09 发布

阿阿阿阿阿夏

最新推荐文章于 2022-07-09 18:08:09 发布

阅读量613

点赞数 4

文章标签： python tensorflow

本文链接：https://blog.csdn.net/wwx_wx/article/details/106013108

版权

（一）引言

最近换了新电脑显卡是GTX1660Ti,可以跑gpu了，研究了两天终于安好了，好多问题出在各种版本不匹配上，这次安装的anaconda3-5.2.0,python3.6.5,tensorflow-gpu=2.1.0,anaconda里边只需要安装一个tensorflow-gpu就可以搞定一切啦。如果你的电脑的显卡支持安gpu上一篇文章那就当练手了吧。那我们就开始吧！

（二）更新驱动

安装所有的东西之前要首先更新驱动。

1下载驱动更新

英伟达驱动更新
进入官网之后设置自己电脑的配置会自动匹配驱动更新以便下载，我是GEFORCE GTX 1660Ti，点击下载
在这里插入图片描述

2安装

下载之后进行安装，安装完成之后桌面图标如下图所示
在这里插入图片描述

3进行驱动更新

点击下载好的驱动（首次打开要注册然后再登录）进入之后如下图所示
在这里插入图片描述
点击左上角的驱动程序，如下图显示

点击下载

下载完成

（三）安装anaconda3-5.2.0

安装包可以从百度分享中进行下载，在这就不详细说anaconda的安装过程了，在上一篇文章中都有详细过程，但是有一点要强调，在安装过程中不要选择path,以免带来后续的问题，安装完成之后不要忘记设置环境变量，上一篇中写了如何设置环境变量。
在这里插入图片描述

（四）安装tensorflow-gpu

1安装tensorflow-gpu环境

anaconda安装完成之后，在电脑的左下角找到Anaconda3(64-bit)
在这里插入图片描述
点击Anaconda Prompt，输入
conda create -n tensorflow-gpu python=3.6.5

输入y之后进行安装

2安装jupyter

要先安装jupyter然后再安装tensorflow-gpu，因为我之前先安的tensorflow-gpu然后安的jupyter会使tensorflow-gpu的版本发生改变。
首先启动刚才安装的tensorflow-gpu环境
activate tensorflow-gpu
在这里插入图片描述
安装ipython
conda install ipython 输入y进行安装

安装jupyter
conda install jupyter 这个安装的东西有点多输入y进行安装
安装完成之后会出现一个DEBUG但是没有太大关系,在电脑左侧找到Anaconda3(64-bit)，找到安装的jupyter
在这里插入图片描述
双击之后可以自己打开（需要自己改变默认位置）
右上角New一个Python3

输入以下代码查看jupyter的默认路径，进入到这个文件夹下打开jupyter_notebook_config.py

找到以上图片位置之后改成下图的样式，单引号里边就是你自己期待更改的目录的位置
在这里插入图片描述
右键单击jupyter Notebook此时还要找到jupyter的位置

右键单击Jupyter Notebook(tensorflow-gpu) 点击属性，需要把双引号里边的东西删除掉，点击确定，此时位置已经改成自己期待的位置。

3安装tensorflow-gpu

输入conda install tensorflow-gpu （会自动配置最新版本）输入y 进行安装
版本：
cudatoolkit 10.1
cudnn 7.6.5
tensorflow 2.1.0
tensorflow-gpu 2.1.0
在这里插入图片描述

4安装pytorch

输入conda install pytorch torchvision cudatoolkit=10.1 -c pytorch （因为以上安装的cudatoolkit=10.1版本）输入y进行安装
在这里插入图片描述
以上所有的东西都已经安装好了

（五）在pycharm中使用tensoflow-gpu

1配置新的pycharm环境

pycharm安装在这就不说了第一篇文章中都有，打开pycharm点击File找到Settings
在这里插入图片描述
找到以下红色框框，点击右侧的设置，点击Add 添加一个新的环境

按照红色框框里边进行操作和设置
Interpreter设置刚才安装的tensorflow-gpu里边的python，下方的Conda executable默认就好了，不用更改，设置好了之后点击确定。
在这里插入图片描述
点击确定之后，红色框框里边会变成刚才设置的环境，点击确定，需要等待一会，pycharm正在设置这个新的环境

等到全部配置完成之后，可以输入以下代码进行检测，并且运行下图结果表示安装成功。