Tensorflow环境安装记录--无法识别GPU的问题

1、镜像:

-i http://pypi.douban.com/simple --trusted-host pypi.douban.com

2、版本信息(红色标注为我电脑的配置信息)

说明:在安装tensorflow-gpu环境时,一定要注意版本信息的对应,否则会出现各种奇葩的问题。

例如,我在安装tensorflow_gpu,由于默认安装的是最新的版本2.4.0,使用pycharm在运行代码时,编译器无法识别GPU设备信息,只能使用CPU来训练网络,结果训练效率极低;

后来,将tensorflow_gpu的版本降为tensorflow_gpu-2.0.0-alpha0后,编译器最终才能正确调用GPU来训练网络。

版本Python 版本编译器编译工具cuDNNCUDA
tensorflow_gpu-2.0.0-alpha02.7、3.3-3.6GCC 4.8Bazel 0.19.27.4.1以及更高版本

CUDA 10.0

(需要 410.x 或更高版本)

tensorflow_gpu-1.13.02.7、3.3-3.6GCC 4.8Bazel 0.19.27.410.0
tensorflow_gpu-1.12.02.7、3.3-3.6GCC 4.8Bazel 0.15.079
tensorflow_gpu-1.11.02.7、3.3-3.6GCC 4.8Bazel 0.15.079
tensorflow_gpu-1.10.02.7、3.3-3.6GCC 4.8Bazel 0.15.079
tensorflow_gpu-1.9.02.7、3.3-3.6GCC 4.8Bazel 0.11.079
tensorflow_gpu-1.8.02.7、3.3-3.6GCC 4.8Bazel 0.10.079
tensorflow_gpu-1.7.02.7、3.3-3.6GCC 4.8Bazel 0.9.079
tensorflow_gpu-1.6.02.7、3.3-3.6GCC 4.8Bazel 0.9.079
tensorflow_gpu-1.5.02.7、3.3-3.6GCC 4.8Bazel 0.8.079
tensorflow_gpu-1.4.02.7、3.3-3.6GCC 4.8Bazel 0.5.468
tensorflow_gpu-1.3.02.7、3.3-3.6GCC 4.8Bazel 0.4.568
tensorflow_gpu-1.2.02.7、3.3-3.6GCC 4.8Bazel 0.4.55.18
tensorflow_gpu-1.1.02.7、3.3-3.6GCC 4.8Bazel 0.4.25.18
tensorflow_gpu-1.0.02.7、3.3-3.6GCC 4.8Bazel 0.4.25.18