二、在Windows下安装TensorFlow GPU版本教程(超级详细哦)

本教程详细介绍了在Windows环境下安装TensorFlow GPU版本的全过程,包括确认显卡CUDA支持、CUDA和cuDNN的安装、配置环境变量、使用Anaconda创建虚拟环境及安装TensorFlow和Keras。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1、确认显卡是否支持CUDA

1.1、查看支持CUDA的显卡

网址:http://developer.nvidia.com/cuda-gpus
在这里插入图片描述

1.2、查看系统信息

在这里插入图片描述

1.3、查看显卡信息

在这里插入图片描述
由上图知,本电脑的显卡为GeForce 940MX,支持CUDA软件,如下图所示:
在这里插入图片描述

2、安装CUDA

2.1、下载并安装CUDA

网址:https://developer.nvidia.com/cuda-downloads
安装步骤如下图所示:
在这里插入图片描述
下载后双击安装,如下图所示:
在这里插入图片描述

2.2、设置解压缩安装程序的暂存目录

在这里插入图片描述

2.3、同意协议并继续

在这里插入图片描述

2.4、选择精简安装选项

在这里插入图片描述

2.5、警告未安装Visual Studio

在这里插入图片描述

2.6、下载CUDA界面

在这里插入图片描述

2.7、确认是否安装此设备软件

注意:在安装之前关闭杀毒软件,否则会出现错误,安装失败。
在这里插入图片描述
如果安装过程中出现了上述的安全信息框,我们直接点击“安装”按钮即可。

2.8、确认安装

在这里插入图片描述

2.9、安装完成

在这里插入图片描述

3、安装cuDNN

3.1、下载cuDNN

网址:https://developer.nvidia.com/cudnn
在这里插入图片描述

3.2、加入会员

在这里插入图片描述
进入注册步骤:
在这里插入图片描述

3.3、进入下载页面

在这里插入图片描述

3.4、查看下载后的文件

在这里插入图片描述

3.5、复制到其他目录

在这里插入图片描述

3.6、创建tools目录,并粘贴之前复制的文件

在这里插入图片描述

3.7、已复制/粘贴到tools目录,如下图所示:

在这里插入图片描述

3.8、查看cudnn64_5.dll

在这里插入图片描述

4、将cudnn64_7.dll存放的位置加入Path环境变量

4.1、编辑系统环境变量

在这里插入图片描述

4.2、单击“环境变量”按钮

在这里插入图片描述

4.3、编辑环境变量

在这里插入图片描述

4.4、编辑用户变量

在这里插入图片描述

5、在Anaconda建立TensorFlow GPU虚拟环境

5.1、重新启动“命令提示符”程序,并切换到工作目录

在这里插入图片描述

5.2、在Anaconda建立TensorFlow GPU虚拟环境

输入conda create --name tensorflow-gpu python=3.5 anaconda命令
如下图所示:
在这里插入图片描述
环境加载完成后会出现下面界面:
在这里插入图片描述
安装完成的界面如下:
在这里插入图片描述

5.3、启用TensorFlow GPU虚拟环境

在命令行中输入:activate tensorflow-gpu,执行结果如下:
在这里插入图片描述

6、安装TensorFlow GPU版本

在命令行中输入:pip install tensorflow-gpu,执行结果如下:
在这里插入图片描述

7、安装Keras

在命令行中输入:pip install keras,执行结果如下:
在这里插入图片描述
亲!你还欠我一评价,请给予个好评,一点点码出的文字,很不容易。

评论 30
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值