一、确认显卡是否支持CUDA
1.1、查看支持CUDA的显卡
网址:http://developer.nvidia.com/cuda-gpus
1.2、查看系统信息
1.3、查看显卡信息
由上图知,本电脑的显卡为GeForce GTX 1050,支持CUDA软件,如下图所示:
https://developer.nvidia.com/cuda-gpus#compute
二、安装CUDA
2.1、确定 Tensorflow-GPU CUDA CUDNN的版本
确定要下载的Tensorflow-GPU的版本并查看对应的CUDA CUDNN的版本
网址:https://github.com/promise777/tensorflow-windows-wheel
我这边下载的是1.4.0\py36\GPU\cuda91 cudnn7 avx2
对应的CUDA 9.1.85 CUDNN是7.0.5
2.2、下载并安装CUDA
网址: