Win10 tensorflow-gpu安装
1. 搭建win10+python3.6+vs2017+cuda10.0+pycuda开发环境
上一篇cuda选择的是10.1版本,但是,目前没有找到和10.1匹配的cudnn和tensorflow-gpu版本,所以笔者又重装了cude,改用10.0版本,这里注意环境变量也要修改。
2. 安装cudnn
cudnn下载地址.
我选择的版本是7.4.2
下载完成,解压后将bin,include,lib文件夹里的内容分别拷到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0目录下的bin,include,lib文件夹里面,注意不是拷贝整个文件夹而是里面的文件。
拷贝完成之后,将bin目录和lib/x64目录添加到环境变量的path中,不然会报错找不到dll文件。
3.安装tenserflow-gpu
pip3 install tenserflow-g