Windows10+Anaconda下,安装TensorFlow2.4对应的CUDA和cudnn

时间:2021-6-6 作者:qvyue

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包。

Windows10+Anaconda下,安装TensorFlow2.4对应的CUDA和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。或者将链接地址:

https://developer.download.nvidia.cn/compute/machine-learning/cudnn/secure/8.0.5/11.0_20201106/cudnn-11.0-windows-x64-v8.0.5.39.zip?BceN54N0ka-ochMyRRafoYOADvHU4ph–vqXbn9PXaDoUI08PY3oAqY1N_vyCAmlV_yyliuwB5Th5vn9Cn-vXWlTEXE_ZJu_NAz5swZNLza0sWY8qztuOZgwuIkNV0RpdFuPVI6HTs1ONv7p33ovii1pdv4uo4AykXdQ8LyNauUqIl4BvQFHG7aJ542sef0dA-CNXxy0UHn1sC2DnA

复制到迅雷直接下载。

Windows10+Anaconda下,安装TensorFlow2.4对应的CUDA和cudnn
下载cudnn 8.0.5 for 11.0

第三步,将cudnn-11.0-windows-x64-v8.0.5.39.zip解压后,将cudabin文件夹中的所有dll文件拷贝到anaconda 虚拟环境中的Librarybin文件夹下。

Windows10+Anaconda下,安装TensorFlow2.4对应的CUDA和cudnn
image.png

第四步,运行is_gpu_available()函数,测试安装是否成功。若返回值是True,则说明配置安装成功。

Windows10+Anaconda下,安装TensorFlow2.4对应的CUDA和cudnn
运行is_gpu_available()函数
声明:本文内容由互联网用户自发贡献自行上传,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任。如果您发现有涉嫌版权的内容,欢迎发送邮件至:qvyue@qq.com 进行举报,并提供相关证据,工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。