Windows10+Anaconda下，安装TensorFlow
2021-04-06 本文已影响0人
LabVIEW_Python
Windows10+Anaconda下，安装TensorFlow2.4对应的CUDA和cudnn。
当前（2021年4月4日）最大的问题是：TensorFlow 2.4要求CUDA为11.0版本，CUDA 11.0版本对应的cudnn为cudnn 8.0.x，而anaconda服务器上，只有CUDA 11.0，没有cudnn >= 8.0.x，由此导致不能用一个命令：
conda install cudatoolkit=11.0 cudnn=8.0.4
轻松完成安装，而是需要一步步手动安装，真的期望Anaconda，赶紧升级一下cudnn包。
Anaconda库当前没有cudnn 8.0.x
手动安装的具体步骤：
第一步，运行命令
conda install cudatoolkit=11.0
安装cudatoolkit
第二步，到NVIDIA官网：https://developer.nvidia.com/rdp/cudnn-archive，下载cudnn 8.0.5 for 11.0。或者将链接地址：
复制到迅雷直接下载。
下载cudnn 8.0.5 for 11.0
第三步，将cudnn-11.0-windows-x64-v8.0.5.39.zip解压后，将cuda\bin文件夹中的所有dll文件拷贝到anaconda 虚拟环境中的Library\bin文件夹下。
image.png
第四步，运行is_gpu_available()函数，测试安装是否成功。若返回值是True，则说明配置安装成功。
运行is_gpu_available()函数
