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
在这里插入图片描述找到上面这个就算安装成功啦!

猜你喜欢

转载自blog.csdn.net/weixin_42299076/article/details/88707790