碎点篇——tensorflow gpu 版本安装

返回主目录

 

tensorflow gpu 版本安装

 

0. 需要VS环境，如果没有需要下载安装
　　https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/

　　Visual Studio 安装，除了C/C++ 其他都不勾选

1. 解压cuda9.0，点击解压后文件夹的setup.exe去安装
　　如果兼容性检查没有问题，就选择推荐的精简去一路next
　　如果兼容性检查出现感叹号，就选择自定义里面勾选上development和runtime，一路next

2. 解压cudnn，把里面的lib、include和bin目录里面的三个小文件拷贝到
　　cuda安装的目录对应的文件夹里面去

3. 在C:\Users\(用户名)目录下，创建一个pip文件夹，创建一个pip.ini文件，里面就写:
　　[global]
　　index-url = https://pypi.tuna.tsinghua.edu.cn/simple

4. cmd进入到anaconda的scripts脚本文件夹所在的位置，去使用命令行去安装
　　pip install --upgrade tensorflow-gpu==1.6.0
　　cpu版本：(cpu版本，不用处理 0 - 2 步骤)
　　pip install --upgrade tensorflow==1.7.0

如果电脑不是win10，需要去找到对应版本的cuda9.0和cudnn9.0
　　Cuda9.0的下载地址