Win10下搭建Vs2017+CUDA开发环境

Win10下搭建Vs2017+CUDA开发环境

         最近开始进行机器学习相关的学习,TensorFlow下CPU的计算速度实在是不理想,需要引入CUDA对于GPU的支持,由于已经有了的Vs2017的环境,在此基础上安装CUDA9.1版本碰到了一些小问题,在此跟大家分享其中的一些坑。

        CUDA可以在NVIDIA官网根据自己的操作系统以及需要的版本进行下载,下载后在安装的过程中,建议使用默认安装路径,安装过程中如果出现错误,请不要着急点击关闭按钮,可以在安装的组件中查看是具体哪个组件发生了错误。如果是驱动程序的问题,建议下载NVIDIA Geforce对显卡驱动程序进行更新后重试,如果是提示Visual Studio Imgration出现错误,请先卸载Vs 2017后再安装Cuda(这个问题困扰了笔者3天的时间,在网上找了各种方法尝试都没有用,最后无奈重装了vs2017,如果有更好的解决办法欢迎留言)。

        安装完成后,需要对环境变量进行配置,需要添加的环境变量如下图所示:

Win10下搭建Vs2017+CUDA开发环境


        接下来需要对VS 2017进行配置,本人是最新版的vs2017,在cuda的头文件host.config中定义了对于编译器版本的支持范围,cuda9.1默认支持到VC 140,本人尝试了修改头文件以避免版本检查,不过新版的VC仍然无法编译通过,因此只能使用vs installer 安装了vs2015 V140版本的编译器:

Win10下搭建Vs2017+CUDA开发环境

然后用vs打开工程,在相应的项目上使用VS2015(v140)平台工具集:

Win10下搭建Vs2017+CUDA开发环境

至此即可正常编译(有11个项目编译失败是因为需要DirectX的开发库,本人并不需要因此没有关注,需要的朋友可以自行安装尝试)。

Win10下搭建Vs2017+CUDA开发环境

对于具体的cuda配置相关,可以参考下面的文章:

点击打开链接