当前位置 : 主页 > 编程语言 > c++ >

tensorflow gpu 版本安装

来源:互联网 收集:自由互联 发布时间:2021-06-23
0,需要VS环境,如果没有需要下载安装 https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/ Visual Studio 安装,除了C/C++ 其他都不勾选 1,解压cuda9.0,点击解压后文件夹的setup.exe去安装 如果兼

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

网友评论