琢磨了一天的干货!win10系统tensorflow-gpu 安装教程,最简单暴力!
安装anaconda
网上有很多具体的教程,可以参考链接: B站.
查找适合你机器的cuda、cudnn以及python版本
1.
2.
3.然后在这里查找你需要的版本
安装tensorflow环境
网上很多帖子在介绍时候都说要安装官网那个很大的cuda软件,其实并不用。这里介绍最简单的安装方法:
1.打开anaconda prompt
2.创建tensorflow环境
conda create -n TF python=3.6
这里需要说明的是,你装的python版本需要和前面介绍的版本对应。
3.安装你的机器对应的cuda,cudnn以及tensorflow版本,我的本子是950M,成功适配。这里以cuda9.0以及cudnn7.1和tensorflow1.12.0为例(我的机器旧了,版本较低)
cond activate TF //进入TF环境
conda install cudatoolkit=9.0 //安装cuda包
conda install cudnn=7.1 //安装cudnn
pip install tensorflow-gpu==1.12.0 -i https://pypi.tuna.tsinghua.edu.cn/simple//安装tensorflow
检查tensorflow-gpu是否可用
在TF环境下,运行如下图代码
结果显示为True,那么恭喜你,安装成功!