目录
安装anaconda(python 3.6)
现在从官网上直接下载到的 anaconda(2019年1月15日) 所搭载的 python 版本是3.7,而 tensorflow(2019年1月15日)所支持的 python 版本是3.4、3.5、3.6。因此我们需要在 anaconda 安装后手动将 python 版本调整成 3.6。
下载 anaconda
点击这个下载地址,选择download即可。
调整 python 版本
在 anaconda 的终端中输入
conda install python=3.6
安装成功后,输入 python
检查看看现在的版本对不对。
安装 CUDA 以及 cuDNN 库
tensorflow(2019年1月15日)支持的 CUDA 版本是9.0,因此在下载的 CUDA 以及 cuDNN 库的时候要注意版本的选择。
下载 CUDA 和 cuDNN 库
点击这个下载地址,选择 Windows->10->exe(local)->Base Installer->download,下载CUDA 9.0。
点击这个下载地址,选择 Download cuDNN,登录你的账号,跳转后挑选 I agree,选择 CUDA 9.0 下载 Windows 10 版本的库。
安装 CUDA
由于CUDA 9.0 的安装包中包含了老版本的显卡驱动,当你的显卡较新时,可能出现显卡驱动识别不了你的显卡,没关系,直接无视之,选择“精简”完成安装即可。
可以在命令提示符中输入
nvcc -V
验证 CUDA 是否正确安装
安装cuDNN 库
参照 tensorflow 官网的做法,在C盘根目录中创建 tools 目录,将下载的 cuDNN库 解压放入即可。
添加环境变量
右键我的电脑->属性->高级系统设置->环境变量->Path 选择编辑,再将下列路径逐个加入即可。
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\extras\CUPTI\libx64
C:\tools\cuda\bin
安装 tensorflow
在anaconda的终端中输入
conda create -n tensorflow python=3.6
完成后,激活 tensorflow
activate tensorflow
安装 tensorflow_gpu
pip install tensorflow-gpu
验证安装效果
python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"
结果如图所示
安装 opencv
在 anaconda 终端中输入
conda install -c menpo opencv
即可完成 opencv 的安装。接着在 python 中输入
import cv2
print(cv2.__version__)
验证是否安装成功。