tensorflow环境测试

发表于 更新于 阅读次数：

tensorflow 安装

docker环境 安装

TensorFlow Docker 要求
在本地主机上安装 Docker。
如需在 Linux 上启用 GPU 支持，请安装 NVIDIA Docker 支持。
请通过 docker -v 检查 Docker 版本。对于 19.03 之前的版本，您需要使用 nvidia-docker2 和 –runtime=nvidia 标记；对于 19.03
及之后的版本，您将需要使用 nvidia-container-toolkit 软件包和 –gpus all 标记。这两个选项都记录在上面链接的网页上。

所以需要先安装显卡驱动

  • (网络不一定通,自行使用魔法)

安装 NVIDIA 容器工具包

官网

1
2
3
4
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
  && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
    sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
    sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
1
apt-get install -y nvidia-container-toolkit
1
sudo nvidia-ctk runtime configure --runtime=docker
1
sudo systemctl restart docker

*