Deepin 安装tensorflow gpu亲测教程(适用于双显卡)

本文详述在Deepin Linux 15.9上安装Tensorflow 1.8.0 GPU版的全过程,包括安装Python、切换到大黄蜂显卡驱动、安装CUDA 9.1、cuDNN 7.1.3和NCCL 2.1.15,以及编译安装Tensorflow。步骤清晰,适用于双显卡环境。

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

前言

这两天为了在Deepin Linux上安装tensorflow-gpu版真的是一把辛酸一把泪,好在经过无数次踩坑终于完美解决。以下是我的计算机的配置:

  • 系统:Deepin Linux 15.9 64位
  • GPU:
    • Intel Corporation 4th Gen Core Processor Integrated Graphics Controller (核显)
    • NVIDIA Corporation GK107M [GeForce GT 755M] (独显,-_-||虽然显卡有点辣鸡但gpu版还是比cpu版要快得多)

本文安装过程中使用的相关构建配置如下:

  • Tensorflow: 1.8.0 gpu版
  • Python: 3.6
  • CUDA: 9.1 (Deepin 15.9源内版本CUDA只有9.1)
  • cuDNN: 7.1.3
  • NCCL: 2.1.15
  • 编译器: GCC 4.8
  • 构建工具: Bazel 0.10.0

注意:本文适用于64位的计算机(现在基本都是),安装的是Tensorflow-gpu 1.8.0版,其他版本推荐使用如下图所示的官方经测试的构建配置,安装方法不变。从图中可见,不同tf版本其实也就构建工具版本不同,所以其他工具版本和我保持一致就行。
Tensorflow经过测试的构建配置
在文章的最后会附上我踩过的坑,供大家参考。在15.9的系统和Intel、Nvidia双显卡的环境下,按照本教程应该是可以安装成功的,一步一步做就好。

方法概述: 使用大黄蜂方案(平时用开源,需要时手动切换Nvidia驱动独显,稳定省电),安装源内版本的CUDA,手动安装cuDNN和NCCL,使用源代码编译安装Tensorflow-gpu(因为源内CUDA只有9.1,tf官方只有支持9.0的,所以只能自己编译源代码安装)。

Let’s GO

Step 1: 安装Python

推荐安装Anaconda,附带各种方便科研和开发的库的Python发行版,我安装的是Anaconda 3.6。怎么安装Anaconda请看这里。已经安装Python的可略过。

Step 2: 切换到大黄蜂显卡驱动解决方案

使用系统自带的“显卡驱动管理器”,傻瓜切换到大黄蜂解决方案。
切换大黄蜂方案
切换成功后,在命令前加上optirun即可使用独立显卡运行程序。在终端中输入

optirun -b none nvidia-settings -c :8

即可打开NVIDIA X服务器设置,查看显卡及驱动信息,可见我安装上的驱动版本为390.67。CUDA 9.0要求显卡驱动384.xx以上,满足要求。
Nvidia X显卡服务器设置

Step 3: 安装源内版本的CUDA

根据我的尝试,自己安装CUDA 9会不识别显卡驱动,所以此处采用源内CUDA。安装很简单:

sudo apt install nvidia-cuda-dev nvidia-cuda-toolkit libcupti-dev nvidia-nsight nvidia-visual-profiler

一条命令搞定确实是很方便,安装后使用nvcc --version命令即可看到CUDA版本信息为9.1。
nvcc --version
安装是容易&#x

评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值