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
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:tensorflow gpu 版本安装 - Python技术站