linux从零开始安装nvidia驱动和tensorflow

Nicoder

于 2017-06-13 21:42:05 发布

阅读量3.5k

点赞数

分类专栏： linux 文章标签： cuda nvidia tensorflow

本文链接：https://blog.youkuaiyun.com/hejunqing14/article/details/73194795

版权

本文详细介绍了如何在Linux系统中从零开始安装NVIDIA驱动、CUDA 8.0和TensorFlow。包括下载安装包，安装NVIDIA驱动避免kernel not found错误，安装CUDA并设置环境变量，接着安装CUDNN，然后通过Anaconda安装Python环境及h5py库，最后通过pip安装TensorFlow并验证安装成功。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装nvidia驱动和CUDA

下载驱动和CUDA安装包，在官网下载对应版本就行
sudo apt-get install linux-headers-$(uname -r) 或者 linux-headers-generic.否则直接安装会报错 kernel not found
安装 nvidia 驱动，一路accept和yes
安装 CUDA，一路yes。安装路径：/usr/local/cuda-8.0/。是否安装推荐的驱动, no
最后显示类似下面Summary内容，表示安装成功。
在/etc/profile中添加：

export PATH=/usr/local/cuda-8.0/lib64:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

保存后

# source /etc/profile
# nvcc -V  检查CUDA
# apt-get install cmake 安装cmake
# cd  /usr/local/cuda-8.0/samples
# make  测试CUDA

测试时间较长，一段没有error即可中止

===========
= Summary =
===========

Driver:   Not Selected
Toolkit:  Installed in /usr/local/cuda-8.0
Samples:  Installed in /storage/installers/cuda_samples, <

最低0.47元/天解锁文章