Ubuntu18.04 tensorflow-gpu环境配置
TensorFlow 官方文档: TensorFlow 2.0.0 安装
准备安装文件
- 下载Nvidia显卡驱动:NVIDIA驱动版本选择
- 下载CUDA:CUDA版本选择,选择要下载的版本,在‘select target platform’中依次选择Linux, X86_64, Ubuntu, 18.04, deb(local). 我的安装经验发现,runfile(local)安装过程看起来步骤少一些,但实际上一直报错无法安装。
- 下载与CUDA对应的CuDnn版本:Cudnn版本选择
安装Nvidia驱动
-
一般下载的安装文件是放在~/Downloads路径下, 通过
cd ~/Downloads
到达安装文件所在的路径. -
安装依赖项:
sudo apt-get install gcc g++ make
-
禁止nouveau
➢ 打开要编辑的文件: