深度学习TensorFlow—GPU2.4.0版环境配置,一文简单易懂详细大全,CUDA11.0、cuDNN8.0
创始人
2024-05-16 04:59:10
0次

深度学习TensorFlow—GPU2.4.0版环境配置,一文简单易懂详细大全,CUDA11.0、cuDNN8.0

前提:电脑拥有英伟达独立显卡!!!,并且安装了anaconda!!!

前提:电脑拥有英伟达独立显卡!!!,并且安装了anaconda!!!

前提:电脑拥有英伟达独立显卡!!!,并且安装了anaconda!!!

本文简单易懂,现在的电脑,一般都是显卡驱动版本超过450的,在Anaconda Prompt中输入查看

nvidia-smi

比如我的查看显示驱动版本是496.76,CUDA支持版本最高到11.5,根据下面的GPU版本和CUDA、cuDNN对应版本,我选择了安装 tensorflow_gpu-2.4.0 版本。

在这里插入图片描述

一、TensorFlow各个GPU版本CUDA和cuDNN对应版本整理(部分):

在这里插入图片描述

二、安装流程:

1、首先安装上表所示的支持环境, Compiler (编译器)和 Build tools (构建工具):

需要官网下载MSVC2019,进行安装后,自带 Build tools 。

点击社区版下载,然后一直点击下一步。进行安装。

在这里插入图片描述

安装过程中出现如下需要选择工作负荷,可以先只勾选Python开发,然后点击安装,后续需要可再进行修改。

在这里插入图片描述

2、CUDA11.0以及cuDNN8.0下载安装本地文件安装:

选择11.00版本的CUDA,按照点击步骤进行下载到本地。

在这里插入图片描述

在这里插入图片描述

然后双击安装CUDA11.0:

可以现自定义一个临时空间,用于解压,用完便自动删除

在这里插入图片描述

选择自定义安装,取消勾选Visual studio intergration 和Display Driver

在这里插入图片描述

一直下一步,安装完成

下载cuDNN,cuDNN就相当于CUDA的补丁文件,下载时需要登陆账号,网络访问慢的,可以下载我的上传的百度网盘文件:

在这里插入图片描述

下载后解压,然后将文件夹复制到cuda安装文件夹里:

在这里插入图片描述

3、检查CUDA是否配置成功:

打开cmd,输入nvcc -V:,可以看到CUDA版本为V11.0:

在这里插入图片描述

进入此文件路径:(也就是安装的cuda文件夹的路径)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\extras\demo_suite

进入cmd:

分别输入:

.\bandwidthTest.exe.\deviceQuery.exe

最后执行结果为Pass,说明配置好了!!!

在这里插入图片描述

4、anaconda配置虚拟环境Tensorflow24:

打开Anaconda Prompt,创建一个虚拟环境,python版本为3.8:

conda create -n Tensorflow24 python=3.8

进入环境,安装命令:

conda activate Tensorflow24  ##进入环境
pip install tensorflow==2.4.0  -i https://pypi.mirrors.ustc.edu.cn/simple

在这里插入图片描述

安装成功后需要单独安装keras版本与tensorflow版本一致,

pip install keras==2.4.0  -i https://pypi.mirrors.ustc.edu.cn/simple

在这里插入图片描述

5、检查是否安装成功:

python                     #进入python环境
import tensorflow as tf    #导入Tensorflow
tf.test.is_gpu_available()  #查看GPU版本信息,并返回Gpu是否可用
exit()						#退出python环境

在这里插入图片描述

大功告成!@!!!!!!

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...