您的位置:首页 > 其它

windows下tensorflow安装步骤

2017-10-20 15:03 309 查看

无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.0cuDNN 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


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

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息