[人工智能][GAN环境搭建] tensorflow环境搭建

1.docker版tensorflow (推荐)

https://www.tensorflow.org/install

1.1 GPU版

docker pull tensorflow/tensorflow:latest-gpu-py3

docker run -it --name mytensor --rm --runtime=nvidia -p 8501:8501 -p 8888:8888 -p 6006:6006 -v /home/security/anaconda3/nn:/nn tensorflow/tensorflow:latest-gpu-py3 python

将tensorflow_datasets升级到4.3.0

将tensorflow_gpu升级到2.3.0

1.2 非GPU版

docker pull dash00/tensorflow-python3-jupyter

docker run -it -p 8888:8888 -p 6006:6006 -v /$(pwd)/nn:/nn dash00/tensorflow-python3-jupyter

jupyter notebook --allow-root --ip=0.0.0.0

In [1]: from notebook.auth import passwd

In [2]: passwd()

Enter password:

Verify password:

Out[2]: 'sha1:ce23d945972f:34769685a7ccd3d08c84a18c63968a41f1140274'

vim ~/.jupyter/jupyter_notebook_config.py

c.NotebookApp.ip='*' # 就是设置所有ip皆可访问

c.NotebookApp.password =u'sha:ce...刚才复制的那个密文'（如果步骤2忽略，只添加这一行即可）

c.NotebookApp.open_browser = False # 禁止自动打开浏览器

c.NotebookApp.port =8888 #随便指定一个端口

c.NotebookApp.notebook_dir = ""

2.非docker版本tensorflow

概要:

wget https://repo.anaconda.com/archive/Anaconda3-2020.02-Linux-x86_64.sh

bash Anaconda3-2020.02-Linux-x86_64.sh

source anaconda3/bin/activate

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow_gpu==2.2.0

pip install keras==2.2.4

tensorflow/nvidea/cuda:

How to install TensorFlow 2.0 on Ubuntu - PyImageSearch

sudo apt-get update

sudo apt-get upgrade

sudo apt-get install build-essential cmake unzip pkg-config

sudo apt-get install gcc-6 g++-6

sudo apt-get install screen

sudo apt-get install libxmu-dev libxi-dev libglu1-mesa libglu1-mesa-dev

sudo apt-get install libjpeg-dev libpng-dev libtiff-dev

sudo apt-get install libavcodec-dev libavformat-dev libswscale-dev libv4l-dev

sudo apt-get install libxvidcore-dev libx264-dev

sudo apt-get install libopenblas-dev libatlas-base-dev liblapack-dev gfortran

sudo apt-get install python3-dev python3-tk python-imaging-tk

sudo apt-get install libgtk-3-dev

sudo add-apt-repository ppa:graphics-drivers/ppa

sudo apt-get update

sudo apt-get install nvidia-driver-418

sudo reboot now

$ cd ~

$ mkdir installers

$ cd installers/

$ wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux

$ mv cuda_10.0.130_410.48_linux cuda_10.0.130_410.48_linux.run

$ chmod +x cuda_10.0.130_410.48_linux.run

$ sudo ./cuda_10.0.130_410.48_linux.run --override

$ nano ~/.bashrc

# NVIDIA CUDA Toolkit

export PATH=/usr/local/cuda-10.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64

$ source ~/.bashrc

scp ~/Downloads/cudnn-10.0-linux-x64-v7.6.4.24.tgz \

username@your_ip_address:~/installers

$ cd ~/installers

$ tar -zxf cudnn-10.0-linux-x64-v7.6.4.38.tgz

$ cd cuda

$ sudo cp -P lib64/* /usr/local/cuda/lib64/

$ sudo cp -P include/* /usr/local/cuda/include/

$ cd ~

验证GPU:

import numpy as np

import tensorflow as tf

import tensorflow_hub as hub

import tensorflow_datasets as tfds

print("Version: ", tf.__version__)

print("Eager mode: ", tf.executing_eagerly())

print("Hub Version: ", hub.__version__)

print("GPU is", "available" if tf.config.experimental.list_physical_devices("GPU") else "NOT AVAILABLE")

可能的问题：

modulenotfounderror-no-module-named-apt-pkg-error

16.04 - ModuleNotFoundError: No module named 'apt_pkg' error - Ask Ubuntu

You appear to be running an X server; please exit X before installing.

drivers - How to install NVIDIA.run? - Ask Ubuntu

sudo ./NVIDIAxxxx.run --no-x-check

Unknown runtime specified nvidia.

docker启动容器报错 Unknown runtime specified nvidia. - luwanglin - 博客园

docker: Error response from daemon: Unknown runtime specified nvidia.

在docker中使用GPU - 知乎

ImportError: cannot import name 'keras_tensor' from 'tensorflow.python.keras.engine'

https://github.com/tensorflow/addons

Could not load dynamic library 'libcudart.so.10.1'; dlerror

conda install tensorflow-gpu cudatoolkit=10.1

参考：

【深度学习 走进tensorflow2.0】linux 下 tensorflow 2.0 环境安装_赖德发的博客-CSDN博客