windows10安装tensorflow gpu版
2017-12-09 本文已影响511人
Fernando23
本文在参考tensorflow官网安装教程基础上总结
- Step1:安装anaconda
- Step2: 安装Visual Studio
- Step3: 下载安装 CUDA® Toolkit 8.0，将按照目录下 \bin 和 \libnvvp 文件夹添加到环境变量
- Step4: 下载cuDNN，将文件夹下 \bin, \include, \lib三个文件夹下的内容依次复制到cuda安装目录的相应文件夹下
- Step5: 更新显卡驱动
- Step6: 打开anaconda shell，创建包含tensorflow的环境:
conda create -n tensorflow python=3.5
- Step7: 打开tensorflow环境:
activate tensorflow
- Step8: 安装 tensorflow:
pip install --ignore-installed --upgrade tensorflow-gpu
- Step9: 安装完成，用以下语句列出所有tensorflow可用设备
from tensorflow.python.client import device_lib
print( device_lib.list_local_devices())
- Step10: 在anaconda Negivator中单独创建包含tensorflow的spyder，方便书写程序
