windows10下tensorflow安装教程(GPU版本)

作者:朋克周 | 创建时间: 2023-06-22
windows10下tensorflow安装教程(GPU版本)

操作方法

首先是安装python,我这里安装的是python3.6版本

记得添加python的环境变量

激活python3.6的环境:activate python3.6

在cmd中输入pip install tensorflow-gpu

接着安装cuda8.0,在网上下载安装包即可。目前tensorflow只支持cuda8.0,其他版本的都安装不成功。

安装完之后验证一下cuda8.0是否安装成功

接着安装cuDNN5.1forcuda 8.0版本,然后解压生成三个文件夹

分然后将cuda/include、cuda/lib、cuda/bin三个目录复制到到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0对应的include、lib、bin目录下

在系统环境变量CUDA_PATH中添加如下变量: C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin; C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

测试tensorflow-gpu是否安装成功; import tensorflow as tf hello = tf.constant('Hello, TensorFlow!') sess = tf.Session()

温馨提示

个人的分享,希望能帮到大家!
点击展开全文

更多推荐