0. 前言
在安装完tensorflow之后,import tensorflow
之后,会出现这样的提示:cudart64_101.dll not found
,那是因为还没有装CUDA,根据提示,我这里需要去下载10.1这个版本 !
1. 安装CUDA
-
首先去官网
选择合适的版本、系统、下载方式等:
-
下载至本地后,接下来进行安装
双击下载至本地的文件:
在选择了安装位置后:
2. 安装cuDNN
安装完CUDA之后,还需安装对应版本的cuDNN。
cuDNN网址:https://developer.nvidia.com/rdp/cudnn-archive
- 选择对应的版本,系统:
-
需要注册一个账户
注册的这个过程不太顺利,验证时会提示浏览器拦截,解决方法就是换一个浏览器多试几次就能登录成功,填一些信息后就可以下载了;
-
下载完成后,解压
解压后,点击,可以看到其中的文件:
将上边框住的3个文件夹放到
CUDA的目录下:
此时,就算完成了 !
此时,再import tensorflow
就会出现Successfully opened dynamic library cudart64_101.dll
!