初学tensorflow2.X,谈谈安装时踩的坑.
安装tensorflow2
1.方法一:tensorflow版本升到2.x后,直接终端pip install tensorflow已经是默认支持gpu版,中文官网说明https://tensorflow.google.cn/install/gpu?hl=zh_cn
- 特别注意版本对应:tensorflow:2.1,python≥3.5,NVIDIA驱动版本≥418.X,NVIDIA CUDA版本≥10.1,否则后面有各种奇怪问题。
NVIDIA® GPU 驱动程序:CUDA 10.1 需要 418.x 或更高版本。
CUDA® 工具包:TensorFlow 支持 CUDA 10.1(TensorFlow 2.1.0 及更高版本)
cuDNN SDK(7.6 及更高版本)
装完将 CUDA、CUPTI 和 cuDNN 安装目录添加到 %PATH% 环境变量中。例如,如果 CUDA 工具包安装到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1,同时 cuDNN 安装到 C:\tools\cuda,请更新 %PATH% 以匹配路径:
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin;%PATH%
SET PATH=C:\Program Files\NVIDIA GPU Computing Toolki