ubuntu20.04 docker 下编译 tensorflow-gpu

这篇具有很好参考价值的文章主要介绍了ubuntu20.04 docker 下编译 tensorflow-gpu。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

ubuntu20.04 安装tensorflow-gpu

配置:
系统 ubuntu 20.04 LTS
显卡 GTX 1060 6G

1 安装cudatoolkit (我选 CUDA Toolkit 12.2 )

NVIDIA CUDA Installation Guide for Linux
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#prepare-ubuntu

选择 2.7 步骤,下载 deb 包,本地安装
2.7. Download the NVIDIA CUDA Toolkit

https://developer.nvidia.com/cuda-downloads

ubuntu20.04 docker 下编译 tensorflow-gpu,AI,tensorflow,人工智能
选择

linux x86_64 ubuntu 20.04 deb(local)

命令行执行

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.2.1/local_installers/cuda-repo-ubuntu2004-12-2-local_12.2.1-535.86.10-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-2-local_12.2.1-535.86.10-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-2-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

安装完成

2 安装cudNN (我选 CUDA Toolkit 12.2 对应的 版本 cuDNN v8.9.3 )

需要注册并且登录Nvidia 账号
然后到这个地址下载
https://developer.nvidia.com/rdp/cudnn-download

我选择这个,根据CUDA Toolkit 版本选对应的
Local Installer for Ubuntu20.04 x86_64 (Deb):

Download cuDNN v8.9.3 (July 11th, 2023), for CUDA 12.x

然后下载到本地安装

sudo chmod 777 cudnn-local-repo-ubuntu2004-8.9.3.28_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2004-8.9.3.28_1.0-1_amd64.deb

完成

docker 拉取镜像 tensorflow/tensorflow:devel-gpu

参考 Docker Linux 构建 https://tensorflow.google.cn/install/source?hl=zh-cn

在某个目录,我这$PWD/home/wmx/software/tensorDocker

sudo docker run --gpus all -it -w /tensorflow -v $PWD:/mnt -e HOST_PERMS="$(id -u):$(id -g)" tensorflow/tensorflow:devel-gpu bash

报错:

docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].
ERRO[0000] error waiting for container: context canceled 

解决:

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
sudo systemctl restart docker

启动成功:
ubuntu20.04 docker 下编译 tensorflow-gpu,AI,tensorflow,人工智能文章来源地址https://www.toymoban.com/news/detail-647757.html

到了这里,关于ubuntu20.04 docker 下编译 tensorflow-gpu的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • Ubuntu 20.04 中安装docker一键安装脚本

    直接上脚本,依次执行如下命令即可 install docker operation system Ubuntu 18.04+ ways1 : wget https://github.com/grant-tt/docker/blob/main/docker_install.sh bash docker_install.sh ways2: wget http://apollo-pkg-beta.bj.bcebos.com/docker_install.sh bash docker_install.sh

    2024年02月13日
    浏览(60)
  • Ubuntu 20.04 安装docker(傻瓜试一键教程)

    更新 apt 包索引 安装依赖包 添加 Docker 的官方 GPG 密钥: 提示OK后 官方安装  安装最新版本的 Docker Engine-Community 到此安装完成, 可以输入 sudo docker --version 或者 sudo docker run hello-world 测试是否安装成功 如果您想免sudo进入docker的话, 直接输入以下命令,亲测: 以下命令意思就

    2024年02月11日
    浏览(52)
  • Docker笔记--Ubuntu20.04安装Nvidia-Docker

    目录 1--添加软件包存储库 2--安装Nvidia-Docker2 3--测试         安装 nvidia-docker2 软件包并重新加载 docker 守护程序配置:         执行 sudo apt-get install -y nvidia-docker2,可能会出现以下错误:E: Unable to locate package nvidia-docker2,解决方法是完整执行完第 1 步;         重启

    2024年02月16日
    浏览(39)
  • ubuntu20.04如何安装nvidia-docker?

    导入 NVIDIA GPG 密钥 添加 NVIDIA Docker 存储库 安装 nvidia-container-toolkit 检查nvidia-docker是否安装成功 如果 nvidia-docker 安装成功,命令将启动一个包含 NVIDIA GPU 的容器并运行 nvidia-smi 命令,该命令可以显示有关 GPU 状态的信息。输出结果将显示 GPU 型号、显存大小、温度等信息。 如

    2024年02月15日
    浏览(49)
  • 如何在 Ubuntu 20.04 上安装和使用 Docker

    前些天发现了一个人工智能学习网站,通俗易懂,风趣幽默,最重要的屌图甚多,忍不住分享一下给大家。点击跳转到网站。 介绍 Docker是一个可以简化 容器 中应用程序进程管理过程的应用程序。容器允许你在资源隔离的进程中运行应用程序。它们与虚拟机类似,但容器更可

    2024年01月17日
    浏览(46)
  • ubuntu 20.04 docker及nvidia-docker2安装

            使用GPU经常会到要使用nivida-docker2的情况,网上吧老有些不负责人的给安装步骤造成误导,不能一次成功,现将自己一次成功的经验分享如下: 1、安装docker(command by command) # Add Docker\\\'s official GPG key: sudo apt-get update sudo apt-get install ca-certificates curl gnupg sudo install

    2024年01月18日
    浏览(36)
  • Linux(7)Ubuntu20.04 arm64安装Docker

    vi /etc/apt/sources.list 这个命令后面跟了几个软件包的名字,它们分别是: apt-transport-https:这个软件包允许apt使用HTTPS协议来访问软件源。 ca-certificates:这个软件包提供了一些受信任的证书颁发机构的证书,用来验证HTTPS连接的安全性。 curl:这个软件包提供了一个命令行工具,

    2024年02月10日
    浏览(60)
  • Ubuntu20.04 + Docker + ORB-SLAM2

            原生的Docker并不直接支持可视化,需使用x11进行可视化。         OpenCV编译会报ippicv错误,前往链接下载ippicv_2020_lnx_intel64_20191018_general.tgz文件,将其放入3rdparty文件夹,参考文章进行修改,重新压缩文件。         编译完成 --gpus all :使用 NVIDIA GPU 支持。这

    2024年02月20日
    浏览(63)
  • ubuntu 20.04 docker安装emqx 最新版本或指定版本

    1.更新系统包列表: 2.安装Docker: 3.启动Docker服务并设置其开机自启动: 4.确保Docker正在运行: 5.使用Docker命令下载并运行EMQX容器: 如果您需要特定版本的emqx,请将“latest”替换为所需版本的标签,例如“emqx:4.4.3”。 6.确认EMQX容器已经启动: 使用以下命令检查容器状态 如

    2024年02月09日
    浏览(47)
  • Ubuntu20.04内安装Docker并在Docker内安装 ROS Melodic

    目录 一. 在 Ubuntu 20.04 中安装 Docker   1. 阿里云一键安装   2. 手动安装步骤 2.1 卸载老旧版本 2.2 安装环境依赖 2.3 安装 GPG 证书 2.4 设置存储库 2.5 安装 Docker 2.6 验证安装是否成功 2.7 安装 Dockers 网页管理 二. 在 Docker 内安装 ROS(Melodic) 1. 安装 Melodic 2. 构建通用环境依赖 3. 在镜

    2024年01月21日
    浏览(48)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包