linux安装tensorflow-gpu

目录

    • 1. 安装Anaconda
      • 1.1 选择anaconda版本
      • 1.2 上传到远程linux服务器(可选)
      • 1.3 开始进行安装
      • 1.4 配置环境变量
      • 1.5 验证是否安装成功
    • 2. 安装tensorflow
      • 2.1 conda创建专用的环境
      • 2.2 安装tensorflow-gpu=2.2
      • 2.3 检查tensorflow是否正确安装

1. 安装Anaconda

参考

1.1 选择anaconda版本

1.2 上传到远程linux服务器(可选)

直接上传时可能造成文件损坏,可以先将安装文件进行压缩,上传后,在进行解压

在这里插入图片描述

1.3 开始进行安装

执行安装命令,接下来按照提示输入ENTER,yes,注意记录下安装位置(在配置环境变量时会用到的)

bash Anaconda3-5.2.0-Linux-x86_64.sh

linux安装tensorflow-gpu_第1张图片
在提示是否配置环境变量时选择no(后面会手动配置的,如果不想手动配置也可以选择yes)
linux安装tensorflow-gpu_第2张图片

1.4 配置环境变量

# 进入.bashrc文件 (~/表示进入当前用户的主目录)
vim ~/.bashrc

# 按'A'键,进入INSERT模式,在最后面添加下面代码
# 注意:这儿的路径时之前安装anaconda3时的路径
export  PATH="/home/genius01/anaconda3/bin:"$PATH

# 按'Esc'退出INSERT模式
# 输入 :wq 保存更改并退出,(:q是不保存退出)

# 执行刚修改的初始化文件,使之立即生效
source ~/.bashrc

1.5 验证是否安装成功

命令行输入conda,出现帮助信息,则安装成功

2. 安装tensorflow

2.1 conda创建专用的环境

conda activate -n cctf22 python=3.6

# 进入到虚拟环境中
conda activate cctf22

2.2 安装tensorflow-gpu=2.2

  • 自动安装
# 搜索conda支持的tensorflow版本
conda search tensorflow

# 安装tensorflow-gpu=2.2
conda install tensorflow-gpu=2.2

linux安装tensorflow-gpu_第3张图片
linux安装tensorflow-gpu_第4张图片

  • 手动下载安装文件,清华镜像源
    linux安装tensorflow-gpu_第5张图片
    在下载文件所在的目录下

    pip install tensorflow_gpu-2.2.0-cp36-cp36m-manylinux2010_x86_64.whl(下载文件的名称)
    

2.3 检查tensorflow是否正确安装

# 查看tensorflow版本
import tensorflow as tf
tf.__version__
# 是否能够成功启动GPU
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
print('GPU:',tf.test.is_gpu_available())

linux安装tensorflow-gpu_第6张图片

你可能感兴趣的:(系统,软件与环境配置,tensorflow,linux)