最近在学习深度学习,使用TensorFlow。搭建好AlexNet网络后却无法调用GPU,无法打开cudart64_101.dll。因此决定重新安装CUDA和cudnn,以及TensorFlow。
1. TensorFlow与CUDA相对应的关系
原网址及截图如下。我选的是CUDA11.0+cudnn8+TensoFlow-gpu-2.4.0.
链接:https://tensorflow.google.cn/install/source_windows
2. CUDA+cudnn+TensorFlow的安装
有人总结的更好我就直接贴网址了
链接: https://blog.csdn.net/FriendshipTang/article/details/114412291.
3. CUDA安装失败原因
3.1 错误原因
我在安装CUDA是出现了安装错误的情况,以前安装了几次都没遇到过。借用了网图。我遇到的情况,图中版本为11.0.
3.2 解决方法
我按照网上的方法,通过自定义的方式取消Visual Studio Intergration进行安装,但依然没有成功。
我电脑里安装的是vs2010,于是我就想试试安装一下vs2019之后能不能成功。安装完之后果然成功了。
跑AlexNet8的截图
4. 资源地址
链接: TensorFlow-gpu-2.4.1.
链接: CUDA.