[Python知识库]在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

了解本专栏 订阅专栏 解锁全文 学习会员免费看 赠一年