Windows10安装tensorflow-GPU+cuda9.0+cudnn7
感觉很多人都不会在windows上安装tensorflow-gpu版本。
安装gpu的版本首先要确定的是,你电脑的显卡是NVDIA英伟达的显卡不然一切都免谈
首先安装tensorflow-gpu
这一步其实是很简单的,对于有安装Anaconda的人来说实在是再简单不过的
#在命令行中写入
pip install tensorflow-gpu==版本号
#这样就自然而然安装的是tensorflow—gpu版本的啦
下面的这张表格就给去了相对应的版本,版本不对应,最终安装的tensorflow-gpu版本是没有用的
安装cuda
1.首先是下载cuda。
Cuda 下载
小编我选择的是cuda9.0的版本。下载的是1.4GB的cuda
下载完成后,双击运行
安装的时候按照他的默认选择就好,如果想安装在其他的盘可以自行选择。在安装的过程中,电脑安装的杀毒软件会疯狂禁止你的操作,允许操作就行了
经过一系列的安装后,其实环境变量大部分已经自动配好了,我们只需要在PATH路径中在添加
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
下载cudnn7
Cudnn下载
要先登录他的网站才可以进行下载,选择window10的版本就可以下载,无论是7点几的版本都适应。下载后是个压缩包,进行解压
将与之对应的文件复制到相应的目录下即可。
到此为止,tensorflow-gpu就安装完成了