win10下RTX3060配置CUDA,并安装带有GPU支持的tensorflow

news/2025/2/12 19:54:11/

踩过了好多坑之后我只能说网上的大部分教程都太老了,不适合RTX30系显卡,其实正确的方法比那些老方法简单的多,重要的是你显卡是新的,软件版本也一定要新

一、安装CUDA与CUDNN

1首先确保你已经安装了新版的nvidia图形驱动程序。其次你得有C/C++编译环境,没有的话装一个Visual Studio或MinGW。
2 安装CUDA Toolkit 11.3
3 下载CUDNN 11.3,将解压后的全部文件复制到CUDA的安装目录中(默认:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\)。
在这里插入图片描述

4 进入 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\extras\demo_suite中,在命令行运行deviceQuery.exe和bandwidthTest.exe,如果运行后显示pass则表示CUDA及CUDNN成功安装。
在这里插入图片描述

二、安装python3.8

直接在官网下载安装,或用conda安装。

三、安装tensorflow

3.1 安装

一定要用 pip 安装,别用conda
pip install tensorflow
因为新版tensorflow同时包含cpu和gpu支持,所以是tensorflow而不是tensorflow-gpu。别指定2.0或2.3版本,就默认新版的,老版本有bug。

3.2 测试

进入python命令行,执行以下代码,最后显示 “True”即说明已支持gpu。
import tensorflow as tf
print(tf.test.is_gpu_available())
以下是一段来自tensorflow官网的示例代码,看看能否执行成功

import tensorflow as tf
mnist = tf.keras.datasets.mnist(x_train, y_train),(x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test / 255.0model = tf.keras.models.Sequential([tf.keras.layers.Flatten(input_shape=(28, 28)),tf.keras.layers.Dense(128, activation='relu'),tf.keras.layers.Dropout(0.2),tf.keras.layers.Dense(10, activation='softmax')
])model.compile(optimizer='adam',loss='sparse_categorical_crossentropy',metrics=['accuracy'])model.fit(x_train, y_train, epochs=5)
model.evaluate(x_test, y_test)

http://www.ppmy.cn/news/291530.html

相关文章

RTX2060+win10+Tensorflow GPU+CUDA 9.2+CUDNN7.2+python 3.65

RTX2060win10Tensorflow GPUCUDA 9.2CUDNN7.2python 3.65 前言(先别安装Tensorflow,以免后面还要换版本)开始!1、Anaconda3下载与安装2、vs 2015 社区版下载与安装3、CUDA9.2下载与安装4、CUDNN7.2下载与安装 开始最终要的环节!&a…

win10+Tensorflow + cuda +RTX 3090/3080 +cudnn 安装

最近RTX3080 / 3090发售,深度学习计算能力提升巨大,本人第一时间入手进行测试,确实氢弹级别! 使用中,最新的cuda11,cudnn,显卡驱动,tensorflow版本都需要一一对应,而tens…

Win10 + GTX1650显卡 安装 Cuda-10.1 + CuDNN-7.6 + Tensorflow-gpu-1.15.0 一步到位

之前一直使用tensorflow的cpu版本,即便是简单的模型训练增加迭代次数到50次还是有点慢的,所以用gpu版本加速 废话不多说,流程先列出来,有个大致认识:anaconda----Visual Studio 2019----CudaCuDNN----配置环境变量----…

WIN10 + GTX1660Ti配置TensorFlow GPU版本

安装准备: 操作系统: Windows 10Python版本: 3.7.3安装Anaconda3-2019.03-Windows-x86_64自带python3.7.3下载地址:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/参考链接:https://blog.csdn.net/junchengberry/article/details/8…

nvidia 3060 + cuda + cudnn + tf

参考: https://eipi10.cn/deep-learning/2019/11/28/centos_cuda_cudnn/ 1.环境版本: CentOS Linux release 7.8.2003 (Core) Tensorflow-gpu 2.5 nvidia 3060 cuda 11.2.2 cudnn-11.3 2.环境检查: lscpi | grep -i nvidia # 要有nvidia 设…

联想台式机 ubuntu 16.04 安装 Nvidia GTX 1060 3g 驱动

总体过程并不复杂 -屏蔽开源驱动nouveau -关 X-window -运行下载好的驱动程序 但是妥妥整了3个晚上 希望能为后人开路 1、下载驱动文件 和网上大家说的不太一样 建议到这个页面下下载 https://www.geforce.cn/drivers 选择 Geforce 10 series gtx 1060 linux 64-bit c…

手把手教你突破 GPT-4.0 3小时25次的限速!

很多人很郁闷 ,ChatGPT Plus账号在浏览器上使用GPT4.0模型的时候,会受到官方的限制,每3小时只能对话25次,是真的不够用。 但是在手机上使用GPT4.0模型则不会有限制,既然这样,那我们是否也可以在浏览器上无限…

微信头像尺寸大小

微信头像尺寸大小 微信头像大小比例相同即可 微信能传送的文件不能大于10M, 微信公众号图片尺寸;宽度:640px~800px左右,图片大小不超过2M.