win10下 tensorflow gpu版本安装

本文详细介绍了在Windows 10 64位系统上安装TensorFlow GPU版本的全过程,包括配置要求、软件版本选择及具体安装步骤。特别强调了与GPU兼容的软件版本的重要性,如CUDA 8.0、cuDNN v6.0和TensorFlow 1.3.0,以及Anaconda的使用。最后验证了安装的成功。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

刚买了带有GPU版本的电脑,其实想直接在ubuntu下安装,因为有服务器,就不想麻烦了,在windows下装看看。
感觉版本不兼容挺烦的,网上博客不是很靠谱,每个人版本都不一样,最靠谱的还是tensorflow官网 说明。
以下截取了部分:
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
看要求,以为需要安装VS2015,其实不用。

windows10 64bit,gpu是机器学习最入门的配置GeForce GTX 1050 Ti。我选择的是上图windows版本的tensorflow_gpu-1.3.0,python 3.6,cudnn 6,cuda 8。选择安装这个tensorflow版本,就不能安装其他版本的cuda和cudnn了,确实一开始看博客,安装了cuda 9,或者cudnn 5.1,都报错了,一选择这个配置安装后,就一点问题没有。。

步骤:

  1. 安装 Anaconda,这里选的是(Anaconda3-5.0.1-Windows-x86_64)对应的是python3.6,暂时不能下最新的Anaconda,因为它是python 3.7,而tensorflow只兼容到python 3.6。下好了一步步安装就好了,不需要安装在C盘,记得添加到PATH里,就是安装时有让两个前面需要打钩的选项,都要打上勾。下载Anaconda地址
  2. 安装 Cuda,这里选的是cuda 8.0 GA2,下好了一步步安装就好了。下载cuda地址
    在这里插入图片描述
  3. 安装 cudnn,选的是cuDNN v6.0 (April 27, 2017), for CUDA 8.0,下好了,解压后有三个目录,分别添加到cuda路径下的三个目录中去,比如我的cuda安装目录为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,要把cudnn的bin、include、lib中的三个文件分别添加到cuda下的对应目录中去。cudnn 下载地址
  4. 安装 tensorflow,gpu版本,这里选的是tensorflow_gpu-1.3.0,tensorflow,这里选的是tensorflow_gpu-1.3.0,tensorflow,安装方法:cmd到命令行,输入 pip install tensorflow_gpu==1.3.0

安装完毕,可以在cmd中输入 nvcc -V查看cuda是否安装上,然后在spyder或pycharm中 输入 import tensorflow 看是否报错,没报错说明安装成功。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值