在Docker中使用TensorFlow(GPU版+Ubuntu 20.04)

Posted 白马负金羁

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了在Docker中使用TensorFlow(GPU版+Ubuntu 20.04)相关的知识,希望对你有一定的参考价值。

在此前的文章【1】中,我们已经介绍了如何在Ubuntu 20.04中安装Docker。本文将介绍如何在其中使用TensorFlow(GPU)版。

一、安装NVIDIA-Docker

安装完成Docker并检查安装正确(能跑出来hello-world)后,如果需要Docker容器中有GPU支持,需要再安装NVIDIA-Docker,同样参考该项目的官方文档【2】。

Docker-CE on Ubuntu can be setup using Docker’s official convenience script:

Setting up NVIDIA Container Toolkit——Setup the package repository and the GPG key:

 然后,Install the nvidia-docker2 package (and dependencies) after updating the package listing:

$ sudo apt-get u

以上是关于在Docker中使用TensorFlow(GPU版+Ubuntu 20.04)的主要内容,如果未能解决你的问题,请参考以下文章

简略描述在Ubuntu18.03部署Docker+Tensorflow GPU版+Pycharm

tensorflow gpu 在“windows”上没有 docker 服务

python docker 多进程提供 稳定tensorflow gpu 线上服务

Rstudio Server+ Docker 中部署GPU版本的keras+tensorflow

TensorFlow-gpu 似乎没有看到我的 gpu

tensorflow docker gpu 图像未检测到我的 GPU