做机器学习的朋友一定绕不开Tensorflow和Keras,跑模型又会要用到GPU的CUDA框架,其中各种版本兼容问题经常导致刚从一个坑里出来,又立刻掉到另一个坑里.....
于是,踩坑无数的我,总结了一篇教程,希望可以帮助到大家少走些弯路。
1. 确认显卡是否支持CUDA
NVIDIA官网有明确标出支持CUDA的显卡都有哪些,直接去官网查询自己的显卡型号是最高效的(只要在列表里出现了就是支持的),附链接:
CUDA GPUsdeveloper.nvidia.com确认显卡支持就可以进入下一步了;如果显卡不支持,Tensorflow也提供了cpu版本供大家使用(就那个没有-gpu后缀的)
2. 安装VS
CUDA运行的时候需要VS的环境,所以要先安装Visual Studio,下载链接:
Visual Studio IDE、代码编辑器、Azure DevOps 和 App Center - Visual Studiovisualstudio.microsoft.com下载完成,开始安装:
安装完毕:
现在可以开始准备安装CUDA了