linux安装tensorflow-gpu

人工智能92

目录

*
- 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服务器（可选）

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

linux安装tensorflow-gpu

; 1.3 开始进行安装

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

bash Anaconda3-5.2.0-Linux-x86_64.sh

linux安装tensorflow-gpu
在提示是否配置环境变量时选择no（后面会手动配置的，如果不想手动配置也可以选择yes）
linux安装tensorflow-gpu

1.4 配置环境变量


vim ~/.bashrc

export  PATH="/home/genius01/anaconda3/bin:"$PATH

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 search tensorflow

conda install tensorflow-gpu=2.2

linux安装tensorflow-gpu
linux安装tensorflow-gpu

  • 手动下载安装文件，清华镜像源
    linux安装tensorflow-gpu
    在下载文件所在的目录下
pip install tensorflow_gpu-2.2.0-cp36-cp36m-manylinux2010_x86_64.whl（下载文件的名称）

2.3 检查tensorflow是否正确安装


import tensorflow as tf
tf.__version__

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
print('GPU:',tf.test.is_gpu_available())

linux安装tensorflow-gpu
linux安装tensorflow-gpu

Original: https://blog.csdn.net/qq_33682559/article/details/120310206
Author: 不要焦绿
Title: linux安装tensorflow-gpu