Win10+tensorflow1.14.0+cuda10.0环境配置

经过长时间踩坑,终于在自己电脑上配置好tensorflow1.14.0+cuda10.0,最开始在电脑上安装cuda10.1版本,后来安装tensorflow出现问题,加载不了tensorflow模块,换成cuda10.0,最后安装成功。
本机机型:拯救者Y7000P-1060
1、安装VS2017,VS2017社区版可以免费安装,下载地址VS2017下载;
2、安装annaconda,annaconda的安装可参考:annconda安装
通过conda --version查看annconda版本信息,确定annconda安装成功。
Win10+tensorflow1.14.0+cuda10.0环境配置_第1张图片
3、安装CUDA
查看显卡信息,鼠标右键,可以看到NVDIA控制面板如下图,如果无NVDIA控制面板信息,说明显卡驱动可能未安装或者安装有问题,需要重新安装
可以选择去NVDIA官网手动搜索对应的驱动程序,但是安装最新的驱动的时候经常会出现兼容性问题,导致安装失败,驱动官网下载地址如下:显卡驱动官方下载
更推荐去你电脑对应的官网下载驱动安装,在官网驱动下载的地方输入电脑的序列号查找对应的显卡驱动,联想驱动下载官网地址如下:联想驱官方下载地址
安装完成右键,可以显示NVDIA控制面板,
Win10+tensorflow1.14.0+cuda10.0环境配置_第2张图片
右键进入—系统信息,查看对应的CUDA安装版本
Win10+tensorflow1.14.0+cuda10.0环境配置_第3张图片
这里显示是10.1版本,经测试CUDA10.1可以安装,但是安装的tensorflow无法使用,后面又卸载该安装版本,安装tensorflow支持的10.0版本,关于如何卸载CUDA可以参考:卸载CUDA
关于下载CUDA10.0版本地址:下载CUDA不同版本选择对应的版本下载,下载完后安装。
然后下载cuDNN,下载地址:下载cuDNN不同版本,需要注册下载。
关于CUDA和cuDNN的安装详细过程可以参考:CUDA和cuDNN安装
以上安装完成,通过nvcc -V查看cuda版本,确定安装成功,如下图。
Win10+tensorflow1.14.0+cuda10.0环境配置_第4张图片
4、安装tensorflow-gpu
打开annaconda prompt,创建tensorflow环境,命令如下:

conda create -n tensorflow python=3.6

通过conda info –envs查看刚刚创建的tensorflow环境
查看tensorflow环境
通过activate tensorflow激活tensorflow环境
激活tensorflow环境
通过以下命令安装tensorflow-gpu

pip install --ignore-installed --upgrade tensorflow-gpu

安装完成之后,可以通过查看tensorflow版本,确定tensorflow是否安装成功。通过python进入python,通过import tensorflow as tf导入tensorflow模块,通过tf.__version__查看tensorflow版本,如下图所示:
Win10+tensorflow1.14.0+cuda10.0环境配置_第5张图片
至此,tensorflow-gpu1.14.0安装成功,最后通过quit()退出python,通过conda deactivate退出当前tensorflow环境。

你可能感兴趣的:(环境配置)