注意:windows环境装tf1.3的话,python要装3.5,cuda8.0,cudnn6.0
1. CUDA安装
CUDA官网
去上述网址下载对应的CUDA版本安装程序(~ 1.4 GB),双击执行exe就可以安装。
安装完之后系统变量会自动添加。
测试一下是否安装成功,命令行输入 nvcc -V
,看到版本信息就表示安装成功了。
2.cuDNN安装
cuDNN官网
去上述网址下载对应的cuDNN安装程序(~ 173 MB),注意需要注册英伟达开发者账号
其实这个是一个压缩包,解压放到任何一个目录下就行,然后把其中的bin
目录路径添加到Path
环境变量里。
环境变量设置→右键此电脑→属性→高级系统设置→高级→环境变量
3. Python安装
Python官网
去上述网址下载相应版本exe文件(windows版tensorflow只支持python3.5及以上),双击执行安装即可,注意记得勾选添加环境变量选项
4. TensorFlow安装
命令行输入
pip3 install tensorflow-gpu==1.3.0
5. matplotlib
pip3 install matplotlib
6. opencv
pip3 install opencv-python
7. Pillow
pip3 install Pillow