anaconda虚拟环境安装tensorflow2

安装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编辑器

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值