TensorFlow GPU安装

来源:互联网 发布:最新php漏洞 编辑:程序博客网 时间:2024/06/08 18:21

一、先装 anaconda

采用清华镜像,自己选择版本,不过2.x没有mingw编译器,选择3.x的吧

https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/

二、下载cudn

cudn得去官网了,,,然后下载一个exe的压缩包。貌似我直接从我另一台pc拷贝的已经安装的安装包也可以。。会在你安装的那个盘有,这是默认的C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,将bin目录添加到环境变量。网上说还得编译,后面发现不编译也没事,进入cmd

按nvcc- V注意大小写,看是否出现了安装的版本即可。

三、下载cudnn

cudnn也去官网,,,会让你注册等等,,,一定要下载和cudn版本同号的,切记,然后将解压后的文件放入cudn中,lib,bin,include。

四、安装TensorFlow

安装TensorFlow,在线和离线。。。在线比较慢,通常改变pip源,
大概这个路径:C:\Users\lenovo

文件内容:

[global]  
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]  
trusted-host=mirrors.aliyun.com

在线下载:就是 pip install tensorflow-gpu

离线下载:https://www.lfd.uci.edu/~gohlke/pythonlibs/,先去这个网站下载whl,然后通过pip install 路径+文件名进行安装,快很多

测试:import tensorflow不报错就ok


原创粉丝点击