之前搭建过CPU版本的TensorFlow,这次尝试用台式电脑搭建GPU版本的TensorFlow。
电脑配置:操作系统: Win10;
GPU :Nvidia GeForce GTX 1070;
CUDA :10.0版本(查看方式:右键NVIDIA控制面板->帮助->组件)
在TensorFlow官网中查看安装GPU版本对应的计算能力及CUDA的要求:
CUDA版本查看,版本为10.0
在网址中(https://developer.nvidia.com/cuda-gpus)查询对应GPU的计算能力,显卡的计算能力满足TensorFlow GPU版本的要求。
1、安装Python 3.6 使用