前言:本人一开始用的TensorFlow是windows下cpu版本,但是后面训练需求,需要安装tensorflow-gpu版本。为实现tensorflow-gpu花了一个多礼拜的时间,查看了网上好多博客,有的是坑,有的又不详细,走了很多弯路,中途重装了一遍系统。本人实现完成后希望后面的弟兄们少走弯路。
显卡:Geforce GTX 1080Ti
软件:windows10
VS2015(一定要这个,因为要写文件需要用vs的格式)
Anaconda
CUDA10.0(注意:在官网上下载的cuda10,是自带驱动的,我下载的是自带411.31的驱动,如果自己下载驱动,可能会不匹配)
cuNDD 7.3.1
1,首先windows10和vs2015安装就不要我说了,照着教程正常安装就行。
2,Anaconda3安装
2.1Anaconda在官网就可以下载,网址:https://www.anaconda.com/download/ ,官网上可以选择各个操作系统的安装包。