ubuntu tensorflow-gpu+cuda+cudnn安装配置
tensorflow笔记
系统及配置
系统为Ubuntu18.04,GPU为GeForce 940M
推荐安装最新版本驱动。Nvidia已经提供了CUDA、cudnn的deb文件或者runfile,极不推荐使用解压安装。
安装CUDA
须知:CUDA与cudnn、tensorflow有严格的版本对应关系。
这里安装CUDA 10.0,cudnn7.5,tensorflow-gpu 1.13.1
推荐安装CUDA 10.0
CUDA下载链接
推荐下载runfile,下载后添加可执行权限,运行。如已安装Nvidia驱动,可不选驱动安装。安照提示安装即可。推荐安装samples用来测试是否安装成功。
添加环境变量:
sudo gedit ~/.bashrc
在文件最后添加
export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}