windows下tensorflow安装步骤

无GPU

python3 -m pip install tensorflow

一行命令即可,然后在python脚本下测试

python3
import tensorflow

没有报错的话则安装成功

有GPU

python3 -m pip install tensorflow-gpu

安装CUDA与cuDNN

windows下tensorflow不支持CUDA 9.0,这里下载CUDA 8.0与cuDNN 6.0,下载cuDNN时需注册登录
CUDA使用默认安装目录,cuDNN下好后解压,将三个包中的文件放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下对应的文件夹中
具体安装目录参考:CUDA文档与cuDNN文档

测试

python3
import tensorflow

没有报错的话则安装成功

安装jupyter notebook

python3 -m pip install jupyter notebook
jupyter-notebook

浏览器出出现如下图所示则安装成功:

windows下tensorflow安装步骤_第1张图片

你可能感兴趣的:(windows下tensorflow安装步骤)