Windows下GPU版Tensorflow环境搭建

1.安装3.5以上Python            

        相关教程太多啦,这里就略了。

2.安装numpy

        在命令行中执行 pip3 install numpy 即可。

3.使用pip安装TensorFlow

       在命令行中执行 pip3 install --upgrade tensorflow-gpu 即可(请确保你即安装了vc++2015运行库)

4.确认你使用的是N卡并支持CUDA,从英伟达官网下载CUDA9.0并安装(目前WIN版TensorFlow只支持到CUDA9.0

        地址:https://developer.nvidia.com/cuda-90-download-archive

        下载本地安装包(local),安装过程中建议选择自定义模式,并更改(或记下)安装路径,以免安装之后找不到CUDA在哪里。

5.安装cuDNN

        地址:https://developer.nvidia.com/cudnn

        需要注册后下载,解压后将目录中的文件复制到CUDA安装位置的相应目录下。

6.配置环境变量

        在 控制面板\系统和安全\系统\高级系统设置 的高级选项卡中点击环境变量,将CUDA安装位置如xxx/CUDA/v9.0/bin添加到系统环境变量PATH中。

7.测试

        在python命令行中输入 import tensorflow 并执行,若无报错则安装成功。

8.Enjoy!

        

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值