Win10 GPU版本Tensorflow安装
最近由于练习主要集中在机器学习处理图片的各种算法部分,CPU版本的“Tensorflow”已经无法满足使用需求,所以进行了GPU版本的“Tensorflow”的安装,以下为一些问题的记录。
0、Note
- GPU Tensorflow只支持“Nvidia”显卡,所以“AMD”显卡用户无需尝试
- GPU Tensorflow需要“CADU”和“cuDNN”的支持,出现的问题也就在此,主要是相互间的版本对应及安装后的环境变量的配置问题。
- GPU Tensorflow使用的版本为“1.7.0”,不同版本需要的“CADU”和“cuDNN”版本不同,可自行查阅相关文档,“1.7.0”当前仅支持V9.0的“CADU”。
1、安装CADU
CUDA V9.0下载地址: https://developer.nvidia.com/cuda-90-download-archive
具体根据系统等等信息进行选择。这里的重点是此文成文时“CUDA”最新版本为V9.1,这里使用的V9.0的原因是因为前述提到将安装的“1.7.0”版本的Tendorflow支持的“CUDA”版本为V9.0,所以请对应安装
安装包下载速度不错,