目录
基于NVIDIA GeForce MX150 的Windows10安装TensorFlow-GPU详解
折腾了一天TensorFlow-GPU的坑。看了许多网络安装帖子,试了很多版本都有问题。有时候发现并不是安装过程太难,而是需要静下心来不断试错。把自己当成一个人形调试机,总会有一个办法是完全适合你的。
本篇基于NVIDIA GeForce MX150 安装 CUDA,cuDNN,Python(anaconda)TensorFlow-GPU(Windows10操作系统)
安装顺序: 查看配置环境 —> CUDA Toolkit —> cuDNN —> Visual Studio 2015 Community —> anaconda虚拟环境 —> TensorFlow-GPU —> 测试
一、安装环境
我的电脑配置如下:
1.windows 10 64bit 专业版
2.Anaconda 2018.12 Python3.7(需要降级为python3.6)
3. Visual Studio 2015 Community
4.显卡支持NVIDIA GeForce MX150
5. cuda_9.0.176_win10_network.exe
6. cuDNN v7.0.5 (Dec 5, 2017), for CUDA 9.0
7.TensorFlow-gpu 1.7.0
8.Keras 2.2.4
9.numpy 1.16.2
10.PyQt5 5.12.1
二、安装Visual Studio 2015 Community
先上Visual Studio 2015 Community资源百度网盘 提取码:f1qz (对!就是那么贴心)
安装过程可参考博客
需要注意的是:
安装路径可以自由选择
自定义安装不需要 VS2015 的全部组件,只需要与 C/C++ 相关的组件,所以这里只选择了“Visual C++”,将其它用不到的组件全部取消勾选。
然后。。。请开始你漫长的安装等待吧。。。(喝杯茶,看个啥)
三、安装CUDA Toolkit
需要使用TensorFlow-GPU请先查看电脑是否支持GPU显卡:
1.设备管理器 —> NVIDIA GeForce MX150