tensorflow2.0GPU版本的相关环境配置与安装教程
一、CUDA10.0安装
切记tensorflow2.0GPU对应的cuda版本只能是10.0(10.1也不行!!!),绝对不能下载错。
下载地址:
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
二、cudnn7.6安装
tensorflow2.0GPU对应的cudnn版本必须是7.4.1 及更高版本。
下载地址:
https://developer.nvidia.com/rdp/cudnn-download
三、Anaconda安装
下载地址:
https://www.anaconda.com/distribution/
四、TensorFlow2.0GPU安装
pip install tensorflow-gpu==2.0.0-rc1