Ubuntu 18.04安装Docker版本的TensorFlow 2.0_docker pull 下载tensorflow 2.0-CSDN博客

本文链接：https://blog.csdn.net/csfreebird/article/details/102252950

本文档详细介绍了如何在Ubuntu 18.04上安装Docker，并通过Docker安装TensorFlow 2.0的GPU版本。首先，升级pip到最新版本，接着安装显卡驱动，特别是NVIDIA驱动。然后，安装Docker并设置socks5代理。接下来，安装nvidia-docker支持，确保能够运行GPU版本的TensorFlow。最后，启动Docker容器并运行TensorFlow的Python程序。

摘要由CSDN通过智能技术生成

2.1 pip 升级

TF2.0要求pip必须>19.0. 所以需要升级

pip install --upgrade pip
# 如果遇到错误, ImportError: cannot import name 'main'，可用下面的命令解决
hash -d pip

2.2 安装显卡驱动

可以用Ubuntu自带的安装驱动的方式，但最好先去官方站点搜索合适的版本。可以从这个页面里面搜索，找到最新的版本号. Linux内核中包含了开源的nouveau驱动. 但这个驱动不包含对3D的支持, 所以需要安装nvidia的官方驱动.

2.2.1 检查显卡和驱动

也可以用GUI的方式查看, 在setting->detail里面

2.2.2 安装显卡驱动

打开Software & Update，查看Additional Drivers. 此时可以看到果然用的是Nouveau driver.

在网络合适的情况下（大多数都不），可以选在第一个选项直接安装。重启后启动“Nvidia X Server Settings”程序，在"PRIME Profiles tab"里切换显卡驱动程序。如下图

现在编辑/usr/share/applications/software-properties-gtk.desktop文件。如果在中国，用的却是美国的ubuntu repository，需要添加proxychains用来快速下载驱动程序. 如果用的是中国的ubuntu repository，则不需要添加proxychains.

Exec=proxychains software-properties-gtk

这种图形化安装的方式比较简单只管，缺点是下载驱动的过程中没有进度细节展示。需要通过netstat命令去监控.