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的下载地址

    https://developer.nvidia.com/cuda-90-download-archive

  Cudnn v7的下载地址

    https://developer.nvidia.com/rdp/cudnn-archive

查看英伟达型号与cuda版本号的网址:

  https://developer.nvidia.com/cuda-gpus

    查看:

      CUDA-Enabled GeForce and TITAN Products

    台式机:

      GeForce and TITAN Products

    笔记本:

      GeForce Notebook Products

/////////////////////////////////////////////////////////////////////

1.常见国内镜像源

https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ 清华

http://mirrors.aliyun.com/pypi/simple/ 阿里

http://pypi.douban.com/simple/ 豆瓣

http://pypi.hustunique.com/simple/ 华中理工大学

http://pypi.sdutlinux.org/simple/ 山东理工大学

http://pypi.mirrors.ustc.edu.cn/simple/ 中国科学技术大学

2. 临时生效使用方法

sudo pip install -i http://pypi.douban.com/simple/ flask

3.配置pip文件

  在windows文件管理器中,输入 %APPDATA%

  (如: C:/Users/用户名/AppData/Roaming/pip/pip.ini)

  在该目录下新建pip文件夹,然后到pip文件夹里面去新建个pip.ini文件,输入以下内容:

    [global]

    index-url = https://pypi.tuna.tsinghua.edu.cn/simple

  Linux平台

  在linux环境下的修改方式和在windows环境下修改方式基本相同,这里简单总结一下:

    (1):在用户的家目录下面创建名为.pip文件夹

    (2):在创建好的.pip文件夹中创建名为pip.conf的文件

    (3):在pip.conf文件中输入以下内容,ok!!!

    [global]

    timeout = 6000

    index-url = http://pypi.douban.com/simple

    trusted-host = pypi.douban.com