提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

文章目录

  • 前言
  • 几行命令搞定


前言

提示:这里可以添加本文要记录的大概内容:
安装tensorflow的过程,希望能给像我一样不太熟悉的小朋友们一点帮助

—linux环境下安装tensorflow

提示:以下是本篇文章正文内容,下面案例可供参考

几行命令搞定

创建虚拟环境
conda create -n tensor python=3.6

激活虚拟环境
conda activate tensor

安装pytorch torchvision cuda,注意版本
pip install torch1.5.0+cu92 torchvision0.6.0+cu92 -f https://download.pytorch/whl/torch_stable.html

一般安装tensorflow-gpu1.12.0的时候会自动安装tensorflow1.12.0,但是我安装完检查发现并没有,所以自己又安了一下
pip install tensorflow-gpu1.12.0
pip install tensorflow
1.12.0

运行代码的时候发现有如下图问题:

把numpy换成1.16.4版本

pip uninstall numpy
pip install numpy==1.16.4

更多推荐

linux环境下tensorflow安装