Tensorflow学习:win10安装tensorflow-gpu

本文详细介绍了在Windows 10上安装Tensorflow-GPU的步骤,包括确认NVIDIA显卡、安装CUDA,配置环境变量,下载并安装cuDNN,以及最后卸载CPU版本的TensorFlow并安装GPU版本。
摘要由CSDN通过智能技术生成

Tensorflow学习:win10安装tensorflow-gpu

1、安装CUDA
1)确认自己的笔记本带有NVIDIA的显卡。电脑->属性->设备管理器->显示适配器
在这里插入图片描述
2)下载安装CUDA 。https://developer.nvidia.com/cuda-downloads
在这里插入图片描述
下载的安装包如下:
在这里插入图片描述
安装过程 和安装其他应用一样,直接下一步下一步,中间可能会有闪屏的现象。
3)安装好之后把CUDA安装目录下的bin和lib\x64添加到 Path环境变量中。
bin路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin
在这里插入图片描述
x64路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\lib\x64
在这里插入图片描述

配置环境变量:
在这里插入图片描述
安装测试:cmd

C:\Users\zhzsdiligence>cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Thu_Jun_11_22:26:48_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.0, V11.0.194
Build cuda_11.0_bu.relgpu_drvr445TC445_37.28540450_0

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

2、安装cuDNN
1)cuDNN下载 。https://developer.nvidia.com/rdp/cudnn-archive
在这里插入图片描述
这个下载需要先注册才能下载,网盘下载好的:链接: https://pan.baidu.com/s/1L8m7PDzzKJb457q7FxQzrw 提取码: 1irj
2)解压压缩包,把压缩包中bin,include,lib中的文件 分别拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0目录下对应目录中 。
3)把C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\CUPTI\lib64下的cupti64_2020.1.0.dll拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin。

在这里插入图片描述
3、安装tensorflow-gpu
卸载之前的TensorFlow(cpu)
安装TensorFlow-gup

cmd:

1.pip uninstall tensorflow 
2.pip install tensorflow-gpu
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值