安装Tensorflow-gpu版本

  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
在这里插入图片描述
3. 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版本已经安装成功
在这里插入图片描述

欢迎加入IT技术交流群:861016679,有问题欢迎留言,一起讨论,进步

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

陶陶name

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值