安装tensorflow2
1 打开anaconda
2 创建虚拟环境 conda create -n tf2.1 python==3.7
3 激活刚创建的环境 conda activate tf2.1
4 安装cuda 工具包 conda install cudatoolkit=10.1
5 安装英伟达深度学习软件包 7.6 版本 conda install cudnn=7.6
(如果报错了,可能不支持gpu) 可直接安装tensorflow (用cpu 运行小的学习案例)
6 安装tensorflow pip install tensorflow==2.1 (指定2.1版本)
7 进入python 验证一下是否成功
(tf2.1) root@2sjlun7e8189s-0:/7f98ef86ac854fa6b50098f79190fd5c# python
Python 3.7.0 (default, Oct 9 2018, 10:31:47)
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
2020-11-23 09:04:45.986537: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer.so.6'; dlerror: libnvinfer.so.6: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda/lib64:
2020-11-23 09:04:45.989306: W tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dynamic library 'libnvinfer_plugin.so.6'; dlerror: libnvinfer_plugin.so.6: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: /usr/local/cuda/lib64:
2020-11-23 09:04:45.989340: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:30] Cannot dlopen some TensorRT libraries. If you would like to use Nvidia GPU with TensorRT, please make sure the missing libraries mentioned above are installed properly.
>>> tf.__version__
'2.1.0'
看到2.1.0 说明安装成功# 欢迎使用Markdown编辑器