ubuntu安装tensorflow gpu版本

最新推荐文章于 2023-12-12 23:26:43 发布

owenbb

最新推荐文章于 2023-12-12 23:26:43 发布

阅读量892

点赞数

分类专栏：环境搭建

本文链接：https://blog.youkuaiyun.com/owenfy/article/details/81674661

版权

环境搭建专栏收录该内容

10 篇文章

订阅专栏

本文详细记录了CUDA9.0的安装步骤，包括驱动安装、CUDA Toolkit配置、符号链接设置及cuDNN安装等。此外，还介绍了如何通过pip安装TensorFlow。

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

由于之前用的1.4，cuda8.0，所以这次重装了一下、

先装驱动

再装cuda9.0，下载

sudo sh cuda_9.2.148_396.37_linux.run

Do you accept the previously read EULA，  输入 accept，进入下一步。
INSTALL NVIDIA Accelerates Graphics Driver for Linux-x86_64 375.26，  输入n，进入下一步；
Install the  CUDA 9.0 Toolkit?，  输入y， 进入下一步；
Enter Tookit Location 点击enter， 进入下一步；
Do you want to install a symbolic link at /usr/local/cuda，  输入y，进入下一步；
Install the CUDA 9.0 Samples?  输入y， 进入下一步；
Enter CUDA Samples Location 点击enter， 进入下一步；

内容

export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}