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服务器(可选)

直接上传时可能会损坏文件。您可以先对安装文件进行压缩,上传后再解压。

[En]

The file may be damaged when uploading directly. You can compress the installation file first, and then decompress it after uploading.

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

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/511394/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球