前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >安装Tensorflow-gpu版本

安装Tensorflow-gpu版本

原创
作者头像
陶陶name
发布2022-05-13 08:31:41
7070
发布2022-05-13 08:31:41
举报
文章被收录于专栏:陶陶计算机
  1. **下载cuda**

**链接:**https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

在这里插入图片描述
在这里插入图片描述

如果电脑上本身就有Visual Studio Integration,要将这个取消勾选,避免冲突了

在这里插入图片描述
在这里插入图片描述

点开Driver comonents,Display Driver这一行,前面显示的是Cuda本身包含的驱动版本是411.31

如果你电脑目前安装的驱动版本号新于Cuda本身自带的驱动版本号,那一定要把这个勾去掉。否则会安装失败(相同的话,就不用去取勾了)

在这里插入图片描述
在这里插入图片描述

接下来需要几分钟的时间安装,这是NVIDIA程序已完成的界面

在这里插入图片描述
在这里插入图片描述

打开此路径,查看nvcc.exe

有这个nvcc.exe就说明CUDA安装已成功

在这里插入图片描述
在这里插入图片描述

打开此文件夹,查看有没有cuti64_100.dll

有这个cuti64_100.dll就说明CUPT1已成功

在这里插入图片描述
在这里插入图片描述

b. cuDNN 安装

CDNN官网链接:

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

选择cuDNN for CUDA10.0

在这里插入图片描述
在这里插入图片描述

解压cuDNN

在这里插入图片描述
在这里插入图片描述

解压会出现三个文件夹:

在这里插入图片描述
在这里插入图片描述

将这三个文件夹里面的文件对应的复制到cuda文件下:

(注意这里是将文件下的文件复制到cuda对应的文件夹里面,而不是将文件夹直接替代cuda下的文件夹(这步特别重要))

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

c. PATH配置

查看CUDA环境路径

我的电脑——>属性——>高级系统设置——>环境变量

在这里插入图片描述
在这里插入图片描述

在系统变量中找到PATH

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

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

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

(要是自定义安装路径的话自行修改)

**测试CUDA:**

cmd

nvcc -V

下面的显示说明我们CUDA的版本是10.0

在这里插入图片描述
在这里插入图片描述
  1. TensorFlow 2.0 安装与测试

a. 确认显卡

安装前要确认显卡是不是NVDIA的

命令行

pip install tensorflow-gpu==2.0.0-beta0

在这里插入图片描述
在这里插入图片描述

b. 测试

测试Tensorflow是否安装成功:

步骤:

打开cmd——>ipython——>import tensorflow as t f

import tensorflow as tf

tf.test.is_gpu_available()

显示True,说明gpu版本已经安装成功

在这里插入图片描述
在这里插入图片描述

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

原创声明:本文系作者授权腾讯云开发者社区发表,未经许可,不得转载。

如有侵权,请联系 cloudcommunity@tencent.com 删除。

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档