Windows系统 Tensorflow2 GPU安装

安装Tensorflow-gpu

1、NVIDIA控制面板→系统信息→组件，查看CUDA支持版本，向下兼容。

2、查看Tensorflow2版本适配的CUDA 和cuDNN,版本选择符合显卡要求。

Tensorflow GPU配置

 3、在Anaconda上创建环境，在使用中，其实完全不需要单独安装cuda和cudnn驱动。

（1）直接利用conda安装cudatoolkit、cudnn以及tensorflow-gpu，我本次装的是2.6。

conda install cudatoolkit=11.2
conda install cudnn=8.1
pip install tensorflow-gpu==2.6.0
#如果下载太慢换个源
pip install tensorflow-gpu==2.6.0 -i https://pypi.tuna.tsinghua.edu.cn/simple 

（2）上述用于下载最新版本，更方便可以直接：

conda search tensorflow-gpu          #查找可下载的版本
conda install tensorflow-gpu==2.5    #下载对应版本

就会自动把所需的packages全部下载安装，很省事。

这两种方法都不用安装整个CUDA，不用再单独配置环境变量。

4、测试是否可用

首先激活该环境，python中输入：

import tensorflow as tf
tf.test.is_gpu_available()

返回结果True。

遇到的问题

最开始在我笔记本1660ti显卡上返回结果Flase，问题是：

Could not load dynamic library 'cusolver64_10.dll'; dlerror: cusolver64_10.dll not found

我在环境路径中library/bin中找到cusolver64_11.dll，修改成cusolver64_10.dll再测试就解决了。