百度AI studio配置cuda +cudnn(详细教程)
从无搭建GPU环境经验,尝试一天终于完成,原来一轮训练需要15个小时使用GPU20多分钟。 环境:Ubuntu16.0.4; nvidia-smi 396 ; 安装 :python 3.6 + cuda 9.0 + cudnn7.4.1 + tensorflow-gpu1.12.0 + keras2.2.4 1.注册 百度的AI studio平台 完成任务(两三分钟)可以获得100小时的GPU 地址 https://aistudio.baidu.com/ (创建项目,上传数据集,启动项目很快就能学会使用) 启动环境进入终端 2.开始配置cuda cudnn 少走弯路:先确定cuda 版本,再选择对应版本的cudnn 和 tf; 2.1查看显卡驱动 nvidia-smi 2.2查看驱动对应的cuda版本 地址: https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html 注:显卡驱动是向下兼容的,平台的驱动是396,无法安装cuda10;所以选择cuda9. 2.3.选择cudnn 版本 查看cuda cudnn 对应版本 https://tensorflow.google.cn/install/source#linux 注:最终选择 cudnn 7 好 tf-1.12.0; 3.安装 3.1 下载 cudnn