赞
踩
tensorflow的github发布源:Releases · tensorflow/tensorflow · GitHub
tensorflow中文网:GPU 支持 | TensorFlow
tensorflow-gpu的清华源:Links for tensorflow-gpu
在 Windows 环境中从源代码构建 | TensorFlow
(在该网页下拉到底,可以看到gpu版本对应的cuda版本和cudnn版本)
博主在清华源下载了tensorflow_gpu-2.4.1-cp36-cp36m-win_amd64.whl,
经过测试,在cuda_11.0.2_451.48_win10和cudnn-11.0-windows-x64-v8.0.5.39可以成功运行tensorflow和tf.keras
试过cudnn 8.0.4版本,在训练模型的时候可能会出现以下报错:
Failed to get convolution algorithm,This is probably because cuDNN failed to initialize.
在网上搜到方法都是说是资源限制、内存分配的问题,但是试过代码都不能解决,后来换了cudnn8.0.5问题就解决了!
写在最前,博主整理的一些cuda和cudnn
链接:https://pan.baidu.com/s/1qneu4fhak2rAPabykH5ZqQ
提取码:g75t
(博主的显卡是 GeForce GTX 1080Ti 11G,对应的计算能力是6.1)
Advanced Driver Search | NVIDIA
记得一定要下载standard版本的gpu驱动,不然会出现以下报错:
CUDA Toolkit 11.4 Downloads | NVIDIA Developer
cuDNN Archive | NVIDIA Developer(需要注册账号)
在下载cuda和cudnn的时候,如果网络一直会不稳定中断下载或者要重新下载,请用复制下载链接用迅雷下载会比较高效一点!
安装和配置软件的系统环境详见我的另外两篇博文:
【软件安装】win7安装tensorflow(GPU版)及环境变量设置
【软件安装】win10装anaconda3-4.2.0、python3.5、opencv3.0和tensorflow1.2(gpu)
【软件安装】CUDA安装失败+win10+VS2015+CUDA8.0+CUDAVisualStudioIntegration找不到+Visual Studio Intergration安装失败
CUDA_PATH_V11_0=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0
%CUDA_PATH_V11_0%\lib\x64;
%CUDA_PATH_V11_0%\extras\CUPTI\lib64;
%CUDA_PATH_V11_0%\bin;
%CUDA_PATH_V11_0%\libnvvp;
D:\Users\Leon_PC\Anaconda3\Scripts
因为基本用到的就是这个文件夹里面conda.exe,所以只添加这一个也是可以的,
关于conda.exe的用法,参考我的另外一篇博文:
【Python】Anaconda软件的conda.exe基本使用
博主整理的一些cuda和cudnn
链接:https://pan.baidu.com/s/1qneu4fhak2rAPabykH5ZqQ
提取码:g75t
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。