win10安装VS2015+CUDA9.0与tensorflow-gpu
0 前言
实验室与某车企合作自动驾驶相关的项目,我在其中负责道路车道线识别的工作。
真实驾驶视频数据集由车企提供,数据量很大,为了训练模型,导师给更新了主机:512G固态硬盘+32G内存+GTX1080显卡。体验爽的一匹。
由于之前的电脑都是师兄配置的,现在他毕业了,只好我自己来配。本文主要参考大神“微调”知乎专栏文章;以及“大米饭盖不住四喜丸子”CSDN博客:
https://zhuanlan.zhihu.com/p/29841665
https://blog.csdn.net/weixin_36368407/article/details/54177380
在其基础上,排除一些自己遇到的坑,把过程记录下来。在开始之前想再次提醒大家,在Windows上安装CUDA和TensorFlow有很多坑,对于各种软件版本以及系统设置的要求可谓非常刁钻,很容易就“误入陷阱”。 我在其中就重装了电脑一次,还好是新电脑 没有啥资料。
1 准备工作
简单的说,我们需要以下软件包。安装的过程要按照这个顺序:
- Windows 10
- Python 3.6(anancanda5.2)
- Visual Studio 2015
- CUDA 9.0 (官网默认是10.0 不要用)
- CuDnn 7.0
1.1 Windows 10
现在实验室 基本上都是win10 了 ,必须适应~
1.2 Python 3.6(anancanda5.2)
习惯使用anancanda去管理python,相信我,绝对好用!!!!可以去清华大学的镜像网站下载anaconda:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
注意下载与自己相同的版本,以及anaconda5.2对应的是 python3.6.6 版本。经测试可以完美使用tensorflow~
1.3 Visual Studio
由于英伟达cuda等程序需要编译,要要使用Visual Studio C++ 的运行环境。
一定别装2017!我因为这个重装电脑了。。。都是泪
Visual Studio 2015的话现在最新的版本是Visual Studio 2015 with Update 3,去官网下载免费的Community版本即可。
但是,微软为了推销新版本,把旧版本藏的很深,我找了挺久才找到, 可以按照下图提示去下载Community版本,登录一下即可。
地址:https://visualstudio.microsoft.com/zh-hans/vs/older-downloads/