《深度学习之TensorFlow入门、原理与进阶实战》—2.3 GPU版本的安装方法

2.3  GPU版本的安装方法

　　如果使用GPU版本，在执行pip之后，还需要安装CUDA和CuDNN。

2.3.1  安装CUDA软件包

　　首先来到CUDA官方网站https://developer.nvidia.com/cuda-downloads，单击Windows按钮后，如图2-8所示。

图2-8  CUDA页面

　　根据自己的环境选择对应的版本，.exe安装文件分为网络版和本地版。网络版安装包比较小，执行安装时再去下载需要的安装包；本地版安装包是直接下载完整的安装包。下载完成后正常安装就可以了。

?注意：CUDA软件包也有很多个版本，必须与TensorFlow的版本对应才行。比如TensorFlow 1.0以后，直到TensorFlow 1.5的版本只支持CUDA 8.0。在本书中也是使用的CUDA 8.0版本来做演示的。可以根据链接https://developer.nvidia. com/cuda-toolkit-archive找到更多版本。