在windows7/10系统下安装TensorFlow-GPU版本的方法

本安装方法适用于windows7和windows10,小白请严格按照步骤来,此处我以TensorFlow-gpu 1.8为例来讲解,如果装tensorflow-gpu的新版本,也可以参考以下方法,但请注意须下载更新版本的CUDA和cudnn。

 

第一步:安装vs2015以上的版本,这个百度一下有很多资源,并且官网也是免费下载的,就不赘述了,安装的时候把C++的组件勾上就行。下载地址:https://visualstudio.microsoft.com/zh-hans/vs/

(注意:电脑里必须有C++的组件才可以成功安装CUDA)

第二步:安装匹配TensorFlow版本的CUDA,比如目前TensorFlow-gpu 1.8需要CUDA9.0版本,网址如下:

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

直接按照默认路径和默认选项去安装就可以

(补充:如果不知道当前tensorflow-gpu版本对应的CUDA软件包,请参考:https://tensorflow.google.cn/install/gpu )

 

第三步:下载匹配TensorFlow版本的cuDNN,比如目前TensorFlow-gpu 1.8需要cuDNN v7.0.版本,网址如下:

https://developer.nvidia.com/rdp/cudnn-archive 

下载好了先别动它,等第五步再操作。

 

第四步:添加环境变量

左击桌面的“计算机”图标,依次选择“高级系统设置”--“环境变量”,在“系统变量(S)”中找到Path选项,双击编辑“变量值(V)”:你需要以下的路径是否在Path中,如果没有,就添加对应的路径(一般是最后两个路径没有):

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include

 

第五步:将cuDNN压缩包解压后得到3个子文件夹(bin,lib,include),你需要把这3个子文件夹复制“NVIDIA GPU Computing Toolkit\CUDA\v9.0”文件夹下(系统会提示已经存在,是否合并,选择确定就可以了),如果找不到文件夹的位置,自己在C盘搜索CUDA就能找到了。

 

第六步:在python的命令栏里输入 pip install tensorflow-gpu  (个人建议先安装Anaconda3,然后直接调用anaconda的命令栏)

 

 

第七步:试探性的在python命令栏里import tensorflow as tf,这次一般还不能成功引用,不过没关系,把电脑重启一次再试一次就可以了。

 

  • 0
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 5
    评论
好的,以下是在 Windows 系统安装 TensorFlow-GPU 2.12.0 的步骤: 1. 确认显卡驱动已经安装并支持 CUDA 10.0 或 10.1,这是 TensorFlow-GPU 2.12.0 的要求。 2. 安装 CUDA Toolkit 10.0 或 10.1,下载地址为: * CUDA 10.0:https://developer.nvidia.com/cuda-10.0-download-archive * CUDA 10.1:https://developer.nvidia.com/cuda-10.1-download-archive 安装时选择自定义选项,仅安装 CUDA 核心组件和 CUDA 工具。安装完成后需要设置环境变量: ``` CUDA_HOME = C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0 (或 v10.1) Path = %PATH%;%CUDA_HOME%\bin;%CUDA_HOME%\extras\CUPTI\lib64 ``` 3. 安装 cuDNN 7.4.2 或 7.5.x,下载地址为: * cuDNN 7.4.2:https://developer.nvidia.com/rdp/cudnn-archive * cuDNN 7.5.x:https://developer.nvidia.com/rdp/cudnn-download 将文件解压到 CUDA Toolkit 的安装路径下,例如: ``` C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64 ``` 4. 安装 Anaconda,下载地址为:https://www.anaconda.com/products/individual 5. 创建一个新的 Python 环境: ``` conda create -n tf_gpu python=3.6 ``` 激活环境: ``` conda activate tf_gpu ``` 6. 安装 TensorFlow-GPU 2.12.0: ``` pip install tensorflow-gpu==2.12.0 ``` 安装完成后可以测试是否成功,例如: ``` import tensorflow as tf print(tf.__version__) ``` 如果输出版本号为 2.12.0,说明安装成功。 希望这些步骤可以帮助你安装 TensorFlow-GPU 2.12.0。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值