安装Tensorflow-gpu的正确打开方式

1. 安装Anaconda
在Anaconda官网下载很慢,建议在清华镜像下载,又快又方便
https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
安装过程和其他软件的安装过程基本没区别,不过别忘记在这一步把两项都选上。
还有,
安装路径最好只改前面的C、D盘,后面的路径不要改。在这里插入图片描述
2. 下载CUDA和cuDNN前的准备工作
安装Anaconda后,打开命令提示符,输入pip install tensorflow-gpu,然后会发现一堆错误,不过不要紧,打开如下路径(具体位置以读者安装路径为准,不过后面的路径是一样的)
D:\ProgramData\Anaconda3\Lib\site-packages\tensorflow\python\platform
在上面路径中找到 build_info.py,打开它,会看到如下代码:
********************************************
cuda_version_number = ‘10.0’
cudnn_version_number = ‘7’
msvcp_dll_name = ‘msvcp140.dll’
nvcuda_dll_name = ‘nvcuda.dll’
cudart_dll_name = ‘cudart64_100.dll’
cudnn_dll_name = ‘cudnn64_7.dll’
********************************************
这里说明了你安装的Tensorflow版本所适合的CUDA及cuDNN的对应版本,去官网下载对应的版本就好,我在版本的选择上入的坑…一言难尽…
下面附上我的版本:
Anaconda: Anaconda3-5.3.0-Windows-x86_64
CUDA: cuda_10.0.130_411.31_win10
cuDNN: cudnn-10.0-windows10-x64-v7.6.2.24

3. 下载并安装CUDA和cuDNN
cuDNN的下载地址如下,下载前需要注册一下
https://developer.nvidia.com/cudnn
CUDA的下载地址如下
https://developer.nvidia.com/cuda-toolkit
安装CUDA…
安装结束后,
a、将cuDNN文件夹中的bin、include、lib中的文件复制到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
下对应的bin、include、lib文件夹中
b、将C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64
中的cupti64_100.dll
拷贝到
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin中
c、添加环境变量
在系统变量中添加
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64

4. 安装Tensorflow
千呼万唤*出来…
做完上述准备工作后,打开命令提示符,输入pip install tensorflow-gpu

5. 验证
打开Anaconda中的Jupyter Notebook输入
import tensorflow as tf
Shift+Enter(屏住呼吸)
如果没有任何错误提示,那么恭喜你,Tensorflow安装成功!
如果有(可能是numpy该升级了,可能是pip该升级了,可能是TensorFlow该升级了…等等)
欢迎留言,本人菜鸟一枚,大家一起讨论!

6. 附图
最后附上一张成功案例图
在这里插入图片描述

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值