系统:Ubuntu 16.04.5 LTS 64 位(内核:4.13.1-041301-generic自己装的、4.4.0-141-generic最开始的)
显卡:GTX1070(驱动:随cuda一起安装,没用软件和更新里的384.130)
软件:TensorFlow 1.12.0 + CUDA 9.0 + cuDNN v7.4.2
因为无线网卡驱动的原因,内核我装了4.13.1,后来升到4.16.1又降下来的,
升级内核过程https://blog.youkuaiyun.com/Kim0403/article/details/85545088
cuda9.0安装指南:https://docs.nvidia.com/cuda/archive/9.0/cuda-installation-guide-linux/index.html
一、先确定需要的版本
首先查看TensorFlow已验证配置:https://tensorflow.google.cn/install/source
已经有1.13.0的测试版了,虽然支持Python3.7,但是需要Ubuntu18.04 + CUDA10.0,作为小白,不想尝试太新的东西。
所以打算安装TensorFlow1.12.0,之前自己安装了最新的NVIDIA驱动跟CUDA9.2,但是运行有问题。所以换CUDA9.0。
二、安装CUDA9.0
0.下载runfile
从官网https://developer.nvidia.com/cuda-toolkit-archive下载了cuda_9.0.176_384.81_linux.run
1.安装前的检查
根据9.0官网教程
- Verify the system has a CUDA-capable GPU.
- Verify the system is running a supported version of Linux.
- Verify the system has gcc installed.
- Verify the system has the correct kernel headers and development packages installed.
验证GPU是否具有CUDA功能
验证是否拥有受支持的Linux版本
验证系统上安装的gcc版本
确保安装正确版本的内核头文件和开发包。
以上四条一般不可能有问题。
安装指南建议尽可能使用特定于发行版的软件包,即RPM或Deb软件包。
我下载了ru