Tensorflow 1.2+Ubuntu 16.04+CUDA 8.0+cuDNN5.1 配置方案
发布时间
阅读量:
阅读量
今日刚完成TensorFlow的配置工作,趁此机会将整个流程详细记录下来,既作为个人经验的总结,也为他人提供一个可供参考的指南。
首先需要安装Ubuntu 16.04系统。通过官网获取镜像文件,并使用UltraISO工具制作可启动的USB安装盘,具体操作步骤在此不再赘述。
若要安装TensorFlow,首要条件是安装Cuda和cuDNN库。(当然,如果仅打算使用CPU进行模型训练,则无需安装这两项组件,可直接跳过此步骤进入TensorFlow的安装环节。)
TensorFlow官网提供了多种不同的安装方式,此处选择最为简便的一种:
通过PIP进行直接安装。这种方式的优势在于操作简单、不易出现错误;但其缺点同样明显——由于过于简便,用户无法自行调整依赖库的具体版本。因此,在按照此方法进行安装时,请务必仔细核对标题中所提示的Cuda和cuDNN版本信息,避免因版本不匹配而引发识别失败的问题。
若所要安装的TensorFlow版本并非1.2,则必须前往TensorFlow官方网站查找与目标版本对应的Cuda和cuDNN版本要求,切勿随意下载错误版本。接下来将详细介绍具体的安装流程:
首先进行Cuda的安装:
前往https://developer.nvidia.com/cuda-downloads 网站下载所需的Cuda安装包,并按照官网提供的指令输入相应
全部评论 (0)
还没有任何评论哟~
