Windows10 通过Anoconda安装TensorFlow（GPU）简明教程

最新推荐文章于 2025-03-19 14:13:09 发布

Brilliant_7

最新推荐文章于 2025-03-19 14:13:09 发布

阅读量1.8k

点赞数

分类专栏： TensorFlow 文章标签： win10 tensorflow gpu

本文链接：https://blog.youkuaiyun.com/Briliantly/article/details/82898364

版权

TensorFlow 专栏收录该内容

5 篇文章

订阅专栏

本文详细介绍了如何在Windows环境下安装CUDA9.0和cuDNN7.3.1，以及如何使用Anaconda环境安装TensorFlow-GPU版本。包括下载链接、安装步骤、环境变量设置及验证方法。

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

1. 安装准备

（1）CUDA 9.0
https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64
建议选择 exe(local) 文件下载
在这里插入图片描述
注：在Anoconda中安装的tensorflow目前只支持到CUDA 9.0，更高版本会报错。如有需要可通过Bazel编译tensorflow源码为.whl文件安装
（2）cuDNN 7.3.1
https://developer.nvidia.com/cudnn （需要注册一下）

2. 安装CUDA和cuDNN

（1）安装CUDA
按照提示的步骤，保持默认安装路径安装即可.
唯一要注意的地方就是，cuda中的某些驱动版本可能较低，这时候就不要勾选该安装选项
（2）安装cuDNN
将解压后得到三个文件夹复制到CUDA的安装目录下对应的文件下即可
默认路径为：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

lib
bin
include

（3）验证
win+R->cmd键入nvcc -V，可以看到如下输出，则说明安装成功
在这里插入图片描述

3. 安装TensorFlow

同样通过win+R->cmd打开终端键入
conda create -n tensorflow-gpu python=3.6.6
接着输入
activate tensorflow-gpu
激活刚刚安装完成的环境，然后输入
pip install --ignore-installed --upgrade tensorflow-gpu安装Tensorflow
不需要时通过deactivate tensorflow-gpu关闭该环境

4. 测试

在激活环境的基础上，通过python进入python终端
复制代码到终端中

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
sess.run(hello)

可以看到如下输出

b'Hello, TensorFlow!'

至此，TensorFlow安装完成。可以愉快的玩耍了~~~o(∩_∩)o

在VSCode中搭建TensorFlow可以参见下篇博文：
https://blog.youkuaiyun.com/Briliantly/article/details/82847804