安装tensorflow

  1. 一.背景
  2. 二.安装

【技术干货】Windows – Tensorflow2.3.0-GPU入坑指南(配置CUDA和CUDNN)

一.背景

  1. 虽然从英伟达控制面板可以看到我自己电脑安装了conda,但是我并没有nvcc命令,猜测是之前安装cuda时有问题,于是重新安装一遍并配置环境变量。
  2. 如果直接conda install tensorflow-gpu或者pip install tensorflow都不同程度上出现了一点问题,也就是无法运行tensflow样例程序。

二.安装

  1. 安装CUDA,安装在d:\software\CUDA目录下,并且配置环境变量,将bin及lib\x64配置到环境变量中去。

  2. 安装CUDNN,这个需要注册才能下载,下载好之后把这四个文件
    image-20211001232017792全部复制到安装目录中去,我的目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1

  3. 用conda新建一个python3.6的虚拟环境
    conda create -n tensorflow python=3.6

  4. 然后激活此环境
    conda activate tensorflow
    并且下载tensorflow和keras
    pip install tensorflow
    pip install keras

  5. 如果有报错缺失dll文件,则自己在网上找这些dll文件下载好后放入到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\bin中,最常见的缺失cudart64_110.dll,或者相似名称的.dll文件,这时候就自己找链接下载吧,110.dll文件
    链接:https://pan.baidu.com/s/1n_YrhR9ZYVgvhPu1zPu4Ng
    提取码:3bzh

  6. 测试代码:

    import tensorflow as tf
    a = tf.constant([1.0, 2.0], name='a')   #构建一个二维数组常量
    b = tf.constant([2.0, 3.0], name='b')
    print(a)
    

    结果:

    tf.Tensor([1. 2.], shape=(2,), dtype=float32)
    

欢迎在评论区中进行批评指正,转载请注明来源,如涉及侵权,请联系作者删除。

×

喜欢就点赞,疼爱就打赏