tensorflow

1.安装 

  cpu版本和gpu版本

    cpu版:pip install ...

  gpu版:

    确认显卡算力大小是否符合要求----》》https://developer.nvidia.com/cuda-gpus

    确认nvidia驱动程序版本 cmd>>>>nvida-smi

    pip install tensorflow-gpu==2.1.0

    tf.test.is_gpu_available()

  print(tf.config.list_physical_devices('GPU'))
根据tensorflow-gpu安装cuda_10.1.105_418.96_win10.exe
下载cudnn-10.1-windows10-x64-v7.6.5.32.zip解压到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
配置环境变量
posted @ 2020-09-29 23:23  CeasonCing  阅读(35)  评论(2编辑  收藏  举报