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
配置环境变量