Windows安装tensorflow-GPU版本

本教程详细介绍了如何在Windows 10系统上安装GPU版本的TensorFlow,包括Anaconda、CUDA 9.0、cuDNN 7.0及OpenCV的配置过程。同时提供了安装步骤、验证方法及常见问题解决方案。
摘要由CSDN通过智能技术生成

python相关安装

电脑64位,Windows10

TensorFlow安装(GPU版本)

配置

win10+CUDA9.0+CUDNN7.0+tensorflow1.9

安装

参考博主:https://blog.csdn.net/haomingzi521/article/details/89192307.

1.安装anaconda

1、官网下载链接:https://www.anaconda.com/distribution/.
清华源快速下载:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
在这里插入图片描述
安装之后配置环境:右击‘我的电脑’,选择‘属性’, 然后’高级系统设置‘,然后’环境变量‘,然后选择系统变量中的’Path’变量,选中,点击‘编辑’,点击‘新建’,在其中添加如下三个路径(此路径根据自己安装路径更改):
此路径根据自己安装路径更改
2、验证是否安装成功:win+r, 输入:cmd, 在命令行中输入:conda list
在这里插入图片描述
3、由于我们安装的anaconda3.7中的python版本是3.7的,所以如果想要使用python3.6或者其他版本的话,可以如下:
在命令行中输入:conda create --name py3.6 python==3.6 创建一个python3.6的环境,在anaconda文件中的envs文件夹下,使用命令:conda activate py3.6 这时,anaconda就是python3.6 可以在命令行中输入:python --version查看,输入命令:conda deactivate 关闭python3.6
,使用python3.7

2.安装cuda9.0

官网下载链接:https://developer.nvidia.com/cuda-downloads.
在这里插入图片描述
安装过程中,会有询问是否允许修改关键位置,全部允许。
然后出现如下
在这里插入图片描述
然后选择同意并继续,在‘选项’中,我选择的是‘推荐’,安装全部。
(附:如果你发现左侧直接跳到了‘安装’那里,并且提示‘NVINDIA程序安装失败’,那么应该是你的显卡驱动程序出现了问题,我的解决方法是使用驱动精灵,重装显卡驱动,笔者没有遇到,但是参考博客有写,不知可行否。)

3.安装cudnn7.0

官网下载链接:https://developer.nvidia.com/rdp/cudnn-download.
需要注册才能下载
之前按照链接博主的,安装cudnn7.1,运行基础运算和全连接神经网络的程序都可以运行,只有开始跑卷积神经网络的时候出现崩溃问题,The kernel appears to have died. It will restart automatically”。查找资料后说cudnn版本问题,于是降低版本。
下载版本如下
在这里插入图片描述
解压后得到:
在这里插入图片描述
打开之后得到三个文件夹:
在这里插入图片描述
然后将这三个文件夹中的内容分别复制到cuda9.0安装路径下的对应文件夹中(注意是文件夹中的内容,不是整个文件夹,注意,复制lib时,里面还有一个文件夹x64,需要复制的是在这里插入图片描述
这个压缩包到对应位置)
验证:打开到CUDA9.0安装路径下的\extras\demo_suite文件夹中(本文是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\demo_suite)然后按住‘shift’+鼠标右击,点击’在此处打开powershell窗口‘
输入: .\bandwidthTest.exe 返回Result=PASS
输入:.\deviceQuery.exe 返回:Result=PASS
说明安装成功了。

4.安装tensorflow-gpu

在windows的‘所有程序’中打开安装的Anaconda程序中的Anaconda prompt
在这里插入图片描述
在命令行中使用命令:
安装python3.6 的tensorflow-gpu环境:conda create -n tensorflow-gpu python=3.6 (中间会判断([y/n?])输入y)
在这里插入图片描述
输入:conda activate tensorflow-gpu 来激活tensorflow环境
在这里插入图片描述
更新一下pip工具,输入命令:python -m pip install --upgrade pip
在这里插入图片描述
安装tensorflow-gpu输入:pip install tensorflow-gpu == 1.9.0
(可以使用清华源加速安装pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu ==1.9.0)
在这里插入图片描述
至此基本安装成功。关闭tensorflow-gpu环境,输入:conda deactivate

5.在Jupyter notebook上使用

还是在Anaconda prompt中:
进入tensorflow环境:activate tensorflow-gpu
在这里插入图片描述
然后输入:conda install jupyter
在这里插入图片描述
安装完成后在打开jupyter notebook 就可以运行啦。

注意

运行GPU版本的时候,要注意运行完一个程序,要点击
在这里插入图片描述
不然会运行出错。

OpenCV安装

在Anaconda prompt中:
进入tensorflow环境:activate tensorflow-gpu
在这里插入图片描述
输入:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple opencv-python(此处利用国内源,不用的命令为pip install opencv-python)
在这里插入图片描述
(附:安装其他别的包也是在这里安装,把后面的opencv-python换为要安装的包就可以了)

jupyter notebook目录的更改

另外记录一下jupyter notebook的初始打开目录的更改方式
在cmd中输入:
jupyter notebook --generate-config
然后在安装目录的下找到 .jupyter文件夹
笔者的是安装在c盘:
在这里插入图片描述
在此文件夹中找到:
在这里插入图片描述
打开找到:
在这里插入图片描述
其中E:\python就是你想要更改后的路径。(原本该行是注释掉的,需要取消掉注释#)
然后修改完成。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值