Windows10安装Tensorflow-gpu版本

Windows环境下安装tensorflow,tensorflow-gpu

  • 安装gpu版本需要提前安装CUDA,CUDNN

如果电脑没有gpu,安装cpu版本也是可以的,只是速度会慢很多。cpu和gpu版本之间应该是可以兼容的,默认调用的是gpu版本,建议只安装一个比较不会报错。

Tensor flow cpu版本安装

  1. 安装anaconda(创建一个python的虚拟环境,里面包含有很多完善的库)
  2. 升级pip
python -m pip install --upgrade pip

显示Successfully字样即安装成功

  1. pip更换国内源,换清华的,豆瓣的都可以(这是一次性的换源方法)
pip install SomePackage -i http://pypi.douban.com/simple/
  1. 安装tensorflow
pip install tensorflow

gpu版本安装(确认有gpu)

  1. 安装anaconda

  2. pip升级

  3. pip换源(代码同上)

  4. 安装cuda
    cuda8.0版本下载链接下载之后安装,可以自定义路径,要记住这个路径,后面要用到

  5. 安装cudnn,实际上就是解压之后把对应的文件放在对应的cuda文件夹里面
    cudnn的下载需要先在官网注册账号

cudnn官网 下载cuDNN v5,for CUDA 8.0 就下载这个下载完成后解压,把里面的文件拷到cuda安装路径下的对应文件夹里面就算安装完成
6. 安装tensorflow-gpu 版本很重要,最好安装低版本,兼容性比较好

pip install tensorflow-gpu==1.2.0

显示Successfully字样即安装成功

接下来检验一下有没有安装成功,在命令行输入pip list
在这里插入图片描述找到上面这个就算安装成功啦!

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值