深度学习笔记(二)——Tensorflow环境的安装

本篇文章只做基本的流程概述,不阐述具体每个软件的详细安装流程,具体的流程网上教程已经非常丰富。主要是给出完整的安装流程,以供参考

环境很重要

一个好的算法环境往往能够帮助开发者事半功倍,入门学习的时候往往搭建好环境就已经成功了一半。

在机器学习或者深度学习的设计研究中,人们往往会使用已经有的网络框架来构建网络模型和设计各种识别分类或者生成算法。

主要可以给我们学习和使用的框架这里推荐两个:Tensorflow或者Pytorch。

当然也有很多其它优秀的框架等待我们学习和使用,例如国内有百度的PaddlePaddle飞桨、Caffe等。后面的内容我们首先在Tensorflow或者Pytorch上开展,毕竟这是大家使用最多的两个框架。

下面是一些主流的网络框架及其标志

1、安装Python

安装python想必各位都已经看过网上各种各样的教程了。但是这里还是多说一句,希望大家无论在windows平台还是linux平台上,都安装conda等环境管理工具使用。学习的过程中往往需要复现他人的大量算法,学习别人撰写代码的经验,所以会经常安装不同类型的环境,使用conda能够极大的便捷我们在环境上的使用。关于python的安装版本,在研究19年以前的算法上推荐使用python3.6,近几年的算法则3.7 3.8均可。安装conda时注意系统环境变量的设置。

常用的conda命令和pip安装命令可以参考:

conda常用命令

2、需要一个好用的IDE或者编辑器

这里推荐使用Pycharm或者VSCode,提示一下,Pycharm的专业版对学生可以申请免费使用, 用学信网的学历认证进行申请,每次更新一次。当然,也完全可以使用免费的VSCode

3、对于GPU设备,安装CUDA工具包

如果你的设备有英伟达的CPU设备,那么你可以进一步安装CUDA和cuDNN,CUDA是利用GPU中CUDA进行计算的关键软件和驱动,cuDNN则专门针对深度学习提供了一些开发接口,二者是利用GPU进行深度学习的关键,同时CUDA和cudnn的版本之间存在严格对应关系。

上图是英伟达官网的版本要求,特点要关注表格第一列和第二列。下载对应版本的CUDA和cudnn安装。

考虑实际,往往会安装多个CUDA版本来适应不同版本的深度学习框架,注意安装时在系统环境变量中设置清楚。

对windows用户,安装CUDA前最好安装Visual Studio并安装C++工作负载。新卡安装2019,旧卡安装2017(当然VS得选项并非必要条件,如果以后想深入学习NN算法,那最好先行安装一个。后面的内容默认安装了19,或17的任何一个版本,来对windows提供完整的C++支持

值得注意:在最新的RTX30系列显卡以及之后的显卡中不在支持CUDA11以前的版本。意味着CUDA10.0和10.1,10.2三个大版本将无法直接在新显卡使用,Tensorflow2.4.0以及之前版本,Pytorch1.7.1及以前的版本无法正常使用,如果有需要可以在英伟达官网下载适配新卡的专用包安装使用。

安装cuda前请先明确是否有特定的版本限制,深度学习框架一搬均有特定版本cuda限制。

4、给python安装相关包

在正确安装conda后,这里我们举例构建一个初学Tensorflow的基础环境

第一步,创建虚拟环境并打开虚拟环境:

复制代码
# bash / Shell
conda create -n tensorflow260 python=3.7

conda activate tensorflow260

第二步,安装相关工具包,以下未指定版本,pip将安装最新版本,只适用学习,复现程序时需要安装指定版本的包,这步在环境配置中及其重要,正确安装完整的,对应版本的包是后续研究的基础。

复制代码
# bash / Shell
# 跟新pip
pip install --upgrade pip

# 检查当前环境中包情况,并无任何ERROR和WORNING
pip list

# 直接安装指定版本tensorflow:pip install tensorflow==x.x.x 
pip install tensorflow
# pip会自动安装需要的各种依赖包,下载过慢的话可以选择切换下载服务器(下例为中科大镜像):
pip install tensorflow -i https://pypi.mirrors.ustc.edu.cn/simple/

# 安装TensorBord
pip install tensorbord

#	安装绘图工具
pip install matplotlib

# 安装机器学习常用库
pip install scikit-learn

5、TensorFlow旧版本对CUDA的支持情况

在安装旧版本TensorFlow前需要先安装对应的CUDA和cudnn。

6、检查环境

检查cuda:

复制代码
# 查看cuda版本
nvcc -V

检查cudnn,在cuda安装目录中执行测试程序,例如在路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\extras\demo_suite下执行bandwidthTest.exe和deviceQuery.exe,不报错则说明安装成功。

检查tensorflow对gpu支持

复制代码
# 激活虚拟环境后:
python

import tensorflow as tf
# tensorflow1,如果返回True则说明安装成功
print(tf.test.is_gpu_available())
# tensorflow2,如果列表不为空说明成功
print(tf.config.list_physical_devices('GPU'))

未写完,24/1/10

相关推荐
AI即插即用9 小时前
即插即用系列 | CVPR 2025 MK-UNet: 多核深度可分离卷积,重新定义轻量级医学图像分割
图像处理·人工智能·深度学习·神经网络·计算机视觉·视觉检测
发光的叮当猫9 小时前
什么是梯度
人工智能·深度学习
淡忘旧梦9 小时前
词错误率/WER算法讲解
人工智能·笔记·python·深度学习·算法
Dream Algorithm9 小时前
POS机的机制,以及流量是怎么传送的
笔记
狐579 小时前
2026-01-21-牛客每日一题-静态区间和(前缀和)
笔记·算法
Yu_Lijing10 小时前
基于C++的《Head First设计模式》笔记——状态模式
c++·笔记·设计模式
阳明Coding10 小时前
golang从入门到通天—数据库操作(gorm框架使用)(最简单最详细的golang学习笔记)
笔记·学习·golang
翱翔的苍鹰10 小时前
CIFAR-10 是一个经典的小型彩色图像分类数据集,广泛用于深度学习入门、模型验证和算法研究
深度学习·算法·分类
Kapibalapikapi11 小时前
思考笔记 | 为什么需要“获取CDN后面的真实IP”
笔记·web安全·思考记录
式51611 小时前
大模型学习基础(九)LoRA微调原理
人工智能·深度学习·学习