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博客


![[GAN环境搭建] tensorflow环境搭建 [GAN环境搭建] tensorflow环境搭建](http://www.mshxw.com/aiimages/31/355180.png)
