快速搭建基于Tensorflow-gpu+cuda+cudnn进行深度学习模型训练
·
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')]
欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐



所有评论(0)