最近要用到tensorflow,所以记录一下安装过程。之前在Windows系统上安装过pytorch还是啥,虽然装好了,但是稀里糊涂的,而且安装过程非常艰辛。这次记录一下在Linux系统安装TensorFlow的过程。真的是记录了过程,安装过程中的1-4后面没采用,可以直接看第5.
系统:Unbuntu 20.04
-
一、安装过程:
1、创建conda虚拟环境
感谢网友,按照博客做的,用conda创建虚拟环境,名称为TensorFlow的虚拟环境,用到的python版本号为3.8,这个版本号感觉很重要,因为现在发布的python版本都3.12了,但是安装TensorFlow时,并不是版本越新越好。
- Python 3.12.1, documentation released on 8 December 2023.
conda create -n TensorFlow python=3.8
2、激活虚拟环境
conda activate TensorFlow
3、安装TensorFlow
根据网友建议,清华镜像可能出现一些问题,所以直接用了阿里云
pip install tensorflow-gpu==2.4.0 -i https://mirrors.aliyun.com/pypi/simple/
4、测试安装
python
import tensorflow
exit()
5、安装过程中的问题
上述安装表面看起来没问题,但是运行我的代码的时候(比较旧的代码)还是出了问题。
最终安装的版本是:
conda create -n tf1.14 python=3.7
pip install tensorflow-gpu==1.14 -i https://mirrors.aliyun.com/pypi/simple/
conda install protobuf==3.20.0
运行代码的时候提示要安装protobuf3.20.x及以下的版本,因此加了最后一句。
所以最后的结论就是python的版本不能太高,TensorFlow的版本不能太高,可能和我的代码比较旧有关系?反正给大家参考!
-
二、参考资料
主要参考了以下内容:
1、TensorFlow2.4 GPU版安装教程_tensorflow2.4安装-CSDN博客
2、https://www.tensorflow.org/install?hl=zh-cn
3、https://www.tensorflow.org/install/pip?hl=zh-cn#package-location (TensorFlow官方的安装说明)