TensorFlow2.0-安装个东西怎么就那么难呢

windows下安装Tensorflwo2.0

  • 使用python3.7

硬件支持

保证Nvidia显卡的算力至少在3.0以上(含3.0)

查看Nvidia显卡的算力

  • cmd命令行查看nvidia-smi。程序的默认位置在C:\Program Files\NVIDIA Corporation\NVSMI下的nvidia-smi.exe

    • 为了时能在命令行中使用改命令,需要将C:\Program Files\NVIDIA Corporation\NVSMI加入系统的环境变量中。

    TensorFlow2.0-安装个东西怎么就那么难呢

软件支持

1.NVIDIA驱动程序

  • 要求使用410.x或者更高版本。
    • 低版本需要到官网下载新的驱动包。但是官网下载又太慢,那么就可以使用迅雷:https://blog.****.net/weixin_37044551/article/details/90487906。
    • 也可以在安装TensorFlow2.0-安装个东西怎么就那么难呢GeForce Experience软件,会自动询问是否保持更新。
    • 鲁大师可以检测新的驱动,并进行安装。
  • 无论哪种方式,都要确保驱动版本符合条件。

2.CUDA-10.0版本

conda install cudatookkit=10.0.130

3.cudnn-版本>7.4.1

conda install cudnn=7.6.0

安装Tensorflow2.0-gpu版本

  • 在确保上面的python语言版本,NVIDIA显卡算力,NVIDIA驱动程序,CUDD版本,cudnn版本均正确后,执行

    pip install tensorflow-gpu==2.0.0

  • 一开始常使用conda install tensorflow-gpu==2.0.0但是显示dynamic library cudart64_100.dll;链接失败,所以选择自己制定各个软件的版本号,然后采用pip的方式安装,结果没有问题。

测试

TensorFlow2.0-安装个东西怎么就那么难呢
TensorFlow2.0-安装个东西怎么就那么难呢

执行

tf.test.is_gpu_available()

结果返回TrueTrue说明安装正确。【上述的执行需要一定时间,需要耐心等待。】

其他:

  1. 保证NVIDIA的驱动在410.x以上
  2. 执行conda install tensorflow-gpu==2.0.0
    • 但是这个操作(第一步成功,第二步太慢了),个人尝试出现了问题,所以采用了上述原生的方式进行安装。