安装前准备:
1.显卡支持CUDA
2.python3.5以上。64位(TensorFlow1.2 开始支持 Python 3.6,之前的官方是不支持的)注意:python一定要是64位的,否则安装不上!
3.确保有稳定的网络连接
4.确保pip版本>=8.1(用 pip -V
查看当前 pip 版本,用 python -m pip install -U pip
升级pip )
5.建议安装 Anaconda,因为这个集成了很多科学计算所必需的库,能够避免很多依赖问题
如果满足上述条件,开始安装。
1.开始下载 CUDA 安装包
点击进入下载
选择适合你的版本。
像安装其他软件一样,正常安装。
安装过程:
输入nvcc -V
查看版本信息。如图表示安装成功。
2.安装cuDNN
点击下载
需要注册。然后才能下载。我下载的是这个版本。
下载后是一个压缩包,进行解压。解压放到任何一个目录下就行,然后把其中的bin目录路径添加到Path环境变量里。
这个是我的路径。
在命令提示符中使用pip install tensorflow-gpu
安装tensorflow-gpu
安装结果:
编写程序进行测试遇到了这个问题:
新版本的 TensorFlow会出现此问题,根据 TensorFlow 的发布说明,该版本会使用 AVX 指令,所以可能会在旧 CPU 上不能运行:
使用旧版 TensorFlow,比如你可以使用 pip install tensorflow-gpu=1.5
来安装1.5版本。