1.conda环境安装

直接去Anaconda官网:https://www.anaconda.com/下载

其中Distribution Installers具有很多已经预装好的科学计算库(方便,但是内存占比比较大)

其中Miniconda Installers只有Python解释器和必要库(推荐,需要什么库自己去下载)

下载好了直接install安装,

找到conda自带的命令提示符

查看版本号

conda --version

创建新环境(推荐使用Python3.8版本的,比较适配)

conda create -n tf2 python=3.8

激活环境

conda activate tf2

查看环境下的包依赖

conda list

查看所有环境

conda env list

退出环境

conda deactivate

下载包依赖(可以清华源加速:https://pypi.tuna.tsinghua.edu.cn/simple)

pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple

2.版本配置(Tensorflow官网上有)

没有适配的就选版本低的(向下可兼容原则)

Tensorflow官网:https://tensorflow.google.cn/install/source_windows?hl=zh-cn

3.cuda下载

1.cuda简便下载(直接使用命令装)

conda search cudatoolkit -c conda-forge

找到适合的版本

conda install cudatoolkit==11.2.0 -c conda-forge

等待完成即可

2.官网手动下载(更快)

nvidia官网:https://developer.nvidia.com,下载号就是一个压缩包

在conda环境输入

conda install "包名"

4.cudnn下载

1.命令行安装

conda search cudnn -c conda-forge

conda install cudnn==8.2.0.77 -c conda-forge

2.手动官网下载(更快)

https://developer.nvidia.com/cudnn-archive

conda install "包名"

5.Tensorflow安装

pip install tensorflow==2.13.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

6.验证

输入

python -c "import tensorflow as tf; print('TF Version:', tf.__version__); print('GPUs:', tf.config.list_physical_devices('GPU'))"

输出

[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]

Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐