深度学习实战 Windows10+Anaconda+CUDA+cuDNN+tensorflow_gpu+pytorch+Jupyter notebook/Pycharm+MNIST
深度学习环境部署+实战Anaconda+tensorflow_gpu+CUDA+cuDNN+Jupyter+MNIST
一、简介
此篇教程记录笔者搭建深度学习环境的详细过程+研究MNIST
MNIST问题看作深度学习的“Hello World”
硬件:
笔记本电脑,GPU为NVIDIA GTX1060
软件:
Windows10+Anaconda+tensorflow_gpu-2.1.0+CUDA10.1+cuDNN7.6+Keras2.3.1/Pytorch+Jupyter Notebook/Pycharm+MNIST
二、环境安装
2.1 Anaconda的安装
2.1.1Anaconda下载安装
Anaconda是深度学习必备的一款软件,可方便我们同时创建多个虚拟环境。
官网下载网址:https://www.anaconda.com/products/distribution#Downloads
根据个人情况下载,笔者下的是Windows python3.9。点击安装。
下载安装参考教程:https://blog.csdn.net/fan18317517352/article/details/123035625
2.1.2配置Anaconda源
通常anaconda的默认源在境外,下载速度会非常慢甚至导致网络错误下载包失败,打开Anaconda Powershell Prompt使用以下方法将清华镜像添加到anaconda。
#配置国内清华源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
#显示源地址
conda config --set show_channel_urls yes
示例:
2.1.3 Anaconda相关命令
笔者只列出常用的几个,其它命令可自行百度了解。
#创建虚拟环境
#创建一个名为env_name,python版本为3.7的虚拟环境
conda create -n env_name python=3.7
示例:
#进入虚拟环境
conda activate env_name
#退出当前环境
conda deactivate env_name
#删除环境
conda remove -n 环境名 --all
2.2安装CUDA和cuDNN
- CUDA。用于GPU的一组驱动程序,它让GPU能够运行底层编程语言来进行并行计算。
- cuDNN。用于深度学习的高度优化的原语库。使用cuDNN并在GPU上运行时,通常可以将模型的训练速度提高50%到100%
2.2.1写在前面
tensorflow分为CPU和支持GPU版本。只有NIVIDIA显卡支持tensorflow gpu版本,并且显卡的算力不能太低的,要求其Compute Capability须不低于3.5。笔者的显卡是1060。
官网列出支持的GPU:https://developer.nvidia.com/cuda-gpus
笔者截图显示部分支持的显卡,具体情况可参考上面的网址。
提醒:只有先确定安装的tensorflow版本,相应的Python,Keras,CUDA和cuDNN安装版本即可确定。否则版本不匹配,会报错。
tensorflow官网推荐:https://tensorflow.google.cn/install/source_windows
tensorflow和keras版本对应:
笔者准备安装tensorflow_gpu-2.1.0,故需要安装Python3.7,CUDA10.1,cuDNN7.6,keras2.3.1。
2.2.2下载安装CUDA和cuDNN
CUDA官网下载:https://developer.nvidia.com/cuda-toolkit-archive
cuDNN官网下载:https://developer.nvidia.cn/rdp/cudnn-archive
cuDNN需要注册账号才能下载。
B站安装教程:https://www.bilibili.com/video/BV1ov41137Z8?p=4
B站看到的视频,很详细。仅在此列出安装建议。
2.2.2.1安装tips
1、安装cuda_10.1.105_418.96_win10.exe
文件时,以管理员身份运行
。笔者默认路径安装。
2、CUDA选择自定义安装。组件的新版本如果比当前的版本低,建议就不要勾选。
3、CUDA安装位置。
ps:图1是视频里的,图2是笔者自己的安装路径。
图1:
图2:
4、测试CUDA是否成功安装。
5、安装cuDNN
将下载的压缩包解压
复制这三个文件
粘贴至此。
至此cuDNN已安好。
2.3安装tensorflow_gpu
2.3.1创建虚拟环境
#创建一个名为tf5,python版本为3.7的虚拟环境
conda create -n tf5 python=3.7
列出现存的虚拟环境
conda env list
#激活虚拟环境
conda activate tf5
2.3.2安装tensorflow_gpu-2.1.0
#安装tensorflow_gpu-2.1.0,安装源为豆瓣
pip install tensorflow_gpu==2.1.0 -i https://pypi.douban.com/simple --trusted-host pypi.douban.com
2.3.3pycharm测试是否成功安装tensorflow_gpu-2.1.0
1、建一个空文件夹,用pycharm打开。
2、解释器修改为我们刚建立的anaconda虚拟环境tf5中的python.exe。
3、测试是否安装成功
#网上找的测试代码
import tensorflow as tf
import os
#os.environ['TF_CPP_MIN_LOG_LEVEL']='2'
a = tf.constant(1.)
b = tf.constant(2.)
print(a+b)
print(tf.__version__)
print('GPU:', tf.config.list_physical_devices('GPU'))
print(tf.test.is_gpu_available())
运行测试代码,发现报错。
复制报错代码,在网上搜索,发现是protobuf版本过高。
网上教程:https://blog.csdn.net/weixin_44834086/article/details/125090000
#进入虚拟环境
#卸载高版本的protobuf
pip uninstall protobuf
#安装低版本protobuf
pip install protobuf==3.19.0
此时再运行测试代码即可成功运行。
2.4安装Keras
#进入虚拟环境,安装keras
pip install keras==2.3.1
2.5安装Jupyter Notebook
2.5.1安装
- Jupyter笔记本:运行深度学习实验的首选方法
- notebook是Jupyter Notebook应用生成的文件,可以在浏览器中编辑。
- 笔记本还可以将冗长的实验代码拆分为可独立执行的短代码,这使得开发具有交互性,而且后面的代码出现问题,你也不必运行前面的所有代码。
#进入虚拟环境,安装jupyter notebook
pip install jupyter notebook
#将虚拟环境添加到jupyter中,其中“new”为虚拟环境在jupyter中显示的名称,与虚拟环境名称可不一致,可自行设置
python -m ipykernel install --name= new
2.5.2运行jupyter notebook
1、anaconda自带jupyter notebook,点Launch
。
2、打开的网页中选择我们创建的虚拟环境tf5
。
下面为成功显示:
三、深度学习实例–MNIST数据集
- 用的代码示例为《Python 深度学习》,在P21。

DAMO开发者矩阵,由阿里巴巴达摩院和中国互联网协会联合发起,致力于探讨最前沿的技术趋势与应用成果,搭建高质量的交流与分享平台,推动技术创新与产业应用链接,围绕“人工智能与新型计算”构建开放共享的开发者生态。
更多推荐
所有评论(0)