Windows10下安装原生TensorFlow GPU版

2024年1月5日 155次阅读 来源: Derek

  1. 下载 CUDA 8.01 和 cuDNN v5.11

  2. 安装CUDA

  3. 解压cuDNN到一个你喜欢的位置，复制文件夹的绝对路径并加到 PATH 环境变量去，然后把该文件夹下的bin文件夹的路径也放到PATH里去

  4. 安装Anaconda Python 3.5 version：
    注意windows tf 仅支持 python 3.5

  5. 安装 tf
    为了避免之后使用 pip 安装 tf 报错，请先在cmd运行以下代码

pip install --ignore-installed --upgrade pip setuptools
之后安装tf
pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0-cp35-cp35m-win_amd64.whl
或者
pip install tensorflow-gpu

接来下检验安装是否成功

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

原文作者: filwaline

    原文作者：Derek
    原文地址: https://segmentfault.com/a/1190000008037754
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞