conda命令安装tensorflow-gpu

背景：我的电脑中已经配置好了anaconda、Jupyter Notebook、Jupyter Notebook(tensorflow)-cpu版本，笔记本的GPU为NVIDIA GeForce MX150

原因：CPU版本跑CNN程序太慢了，所以我决定用conda命令再装一个tensorflow-gpu的版本

注意事项：1、安装GPU版本的tensorflow重要的地方在于要装两个包：cudatoolkit和cudnn；

                  2、我们可以使用conda install tensorflow-gpu命令在装GPU版本的同时一键安装这两个包的最新版本；

                  3、但是NVIDIA驱动版本需要与这两个包的版本相匹配，因为系统中依赖GPU驱动的程序比较多，也就是说conda命令一键安装的两个包的最新版本可能会在运行时报错：cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version。

                   4、于是我们先安装cudatoolkit 9.0版本和cudnn 7.x版本，再用conda命令安装GPU版本，亲测正确运行。

步骤：1、打开Anaconda Prompt；

           2、启动tensorflow：activate tensorflow

           3、先安装cudatoolkit 9.0和cudnn：conda install cudatoolkit=9.0

                                                                    conda install cudnn

           4、最后用conda命令安装GPU版本的tensorflow：conda install tensorflow-gpu

 

参考博客：https://blog.csdn.net/carolynlmk/article/details/86630865