win10安装TensorFlow(GPU)简明教程

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重新编译为.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
接着输入
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.csdn.net/Briliantly/article/details/82847804

猜你喜欢

转载自blog.csdn.net/Briliantly/article/details/82898364
今日推荐