第一步(安装TensorFlow)
最新版的TensorFlow 2.7版本好像同时兼容cpu和gpu
打开anaconda prompt,输入:
pip install tensorflow
不要使用conda安装,会有很多问题,而且pip才是最新版2.7
第二步(安装对应版本cuda以及cudnn)
打开如下网址查看对应TensorFlow版本的cuda以及cudnn版本号
https://tensorflow.google.cn/install/source_windows
打开网址
https://anaconda.org/
分别搜索cudaa以及cudnn,找到对应版本点击进入,复制网页中安装语句如:
conda install -c anaconda cudnn
安装完成后即可使用