Linux下通过Anaconda安装CPU版的TensorFlow和Keras

神经网络   2025-01-01 15:33   28   0  

一、创建conda 计算环境

切换到 conda 用户,执行:

conda create -n tensorflow python=3.10

56028_vqon_6960.png

56033_co7w_9724.png

输入y,按回车Enter执行,提示 done 完成:

56134_akow_2002.png

二、激活环境

source activate tensorflow

56205_7ii3_6201.png

三、安装TensorFlow

为确保安装顺利,先将 pip 升级到最新版本,在激活虚拟环境的终端中输入:

pip install --upgrade pip

56335_ilp8_1354.png

使用以下命令安装 CPU 版的 TensorFlow:

pip install tensorflow-cpu -i https://mirrors.tuna.tsinghua.edu.cn/pypi/web/simple/ --trusted-host pypi.tuna.com

58335_edto_1117.png

58344_2xtt_2302.png

58353_a2pv_5114.png

四、验证安装

1、激活conda环境:

打开新的终端窗口,切换到 conda 用户,执行:

conda activate tensorflow

58489_us0s_1991.png

2、进入python:

58554_f5gm_2313.png

输入执行:

import tensorflow

58702_hcpt_8535.png

执行:

import tensorflow as tf
print(tf.__version__)

若没有报错,并且输出相应版本号,则说明 CPU 版本的 TensorFlow 安装成功。

58801_fkuf_7097.png

五、验证Keras

TensorFlow 2.x 开始,Keras 已经被完全集成到 TensorFlow 中,作为 tensorflow.keras 模块。因此,你应该使用 import tensorflow.keras as keras 而不是 import keras。

进入python后执行:

import tensorflow.keras as keras

可能出现以下warning信息:

This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.To enable the following instructions: AVX2 AVX_VNNI FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.

52442_m2xc_9575.png

这个警告是 TensorFlow 在启动时输出的信息,提醒用户一些当前 TensorFlow 二进制文件已经经过优化,以利用充分可用的 CPU 指令集执行性能关键的操作。同时,警告指出在其他操作中,还有额外的CPU指令集(如SSE、SSE2、SSE3、SSE4.1、SSE4.2、AVX2、AVX512F、AVX512_VNNI、FMA)是可以启用的,但目前还没有在TensorFlow编译时启用。

解决方法:在开头加两句

import os
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'

再执行:

import tensorflow.keras as keras
print(keras.__version__)

如果没有报错,并且能够输出版本号,则说明 Keras 正确。

52448_nyv2_4173.png

博客评论
还没有人评论,赶紧抢个沙发~
发表评论
说明:请文明发言,共建和谐网络,您的个人信息不会被公开显示。