Anaconda3安装tensorflow-gpu
记录一下自己安装tensorflow的过程,实在遇到了很多坑
我是win10系统
首先安装gpu版tensorflow要安装CUDA和cudNN
tensorflow、CUDA和cudNN和python的版本一定要先查清楚不然就要像我一样重新安装很多遍
自己电脑显卡支持的cuda版本可以按这个步骤:
桌面右键->NVIDIA控制面板->帮助->系统信息就可以查看适合安装的cuda版本我的显示是10.0.132但是并没有找到那个版本,10.0.130也安不了,但是9.2和9.0版本都可以安装成功
CUDA下载地址:https://developer.nvidia.com/cuda-toolkit-archive
cudNN下载地址:https://developer.nvidia.com/rdp/cudnn-archive
9.2版本的CUDA我选的是7.3.1版本的cudNN
9.0版本的CUDA我选的是7.1版本的cudNN
最后用的是 9.0版本的CUDA和7.1版本的cudNN
下载cudNN的时候就有写明是适合哪个版本的CUDA,照着下早就没问题了
关于cudNN的安装罗嗦几句
下载好后解压,里面有三个文件夹:bin、include、lib
进入文件夹C:->Program Files->NVIDIA GPU Computing Toolkit->CUDA->v9.0
然后把之前解压的三个文件夹里的文件复制到上面所示路径的对应三个文件夹里面即可
如果想检验是不是成功了:C:->Program Files->NVIDIA GPU Computing Toolkit->CUDA->v9.0->extras->demo_suite可以运行里面的demo,成功运行即安装成功。
(PS:Nivida我觉得非常迷,9.2版本的CUDA安装gpu版tensorflow怎么试都不行,但是9.0的普适版就很高)
tensorflow我安装的是1.10.0版本,先安装了Anaconda之后才安装的tensorflow,个人感觉Anaconda是个很强大的工具,功能性很强,用起来也很方便。
tensorflow的语言是python,安装tensorflow之前也一定要看好版本再进行安装,这是一个对照表
个人建议选择python3.5,后续安其他的安装包兼容性也比较高
好的,接下来安装tensorflow:
打开anaconda promt,输入:conda create --name tensorflow-gpu python=3.6
创建一个名为tensorflow-gpu的新环境;
**:activate tensorflow
查看创建的新环境是否被**: conda info --envs
安装指定版本的tensorflow:pip install tensorflow-gpu==1.4.0
安装好后还需要通过Anaconda Navigator进入tensorflow-gpu环境,安装ipython、spyder、jupyter
运行程序时还需要安装什么直接pip install就可以了