当前位置: 首页 > news >正文

Tensorflow(GPU版本配置)一步到位!!!

Tensorflow(GPU版本配置)一步到位!!!

  • CUDA安装
  • CUDA配置
  • Tensorflow配置常见的包

CUDA安装

配置了N次的Tensorflow–Gpu版本,完成了踩坑,这里以配置Tensorflow_gpu 2.6.0为例子进行安装

以下为tensorflow的版本支持的GPU, 点击此处查看,不要问我其他高版本的怎么不行,我也不知道!跟着来吧!!!!

在这里插入图片描述

CUDA11.2.0下载,安装就不说了,自己来!!!

在这里插入图片描述

CUDA配置

CUDNN8.1.0下载, 看着红框!!!

在这里插入图片描述

这是解压后的CUDNN文件目录

在这里插入图片描述

将此目录的3个文件夹复制至CUDA中,默认路径为:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.2

在这里插入图片描述

打开终端【CMD】, 输入nvcc -V, 出现以下界面,好!!!,CUDA你安装成功了!!!

在这里插入图片描述

Tensorflow配置常见的包

这里使用python3.8的版本, 创建一个新的虚拟环境(virtualenv 文件名), 如果没有virtualenv的使用经历,百度吧!!!,你可以的!!!

tensorflow_gpu == 2.6.0
pandas == 1.4.2
jupyter notebook == 6.5.6
scipy == 1.6.1
matplotlib == 3.5.3

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import tensorflow as tfprint(tf.test.is_gpu_available())

如果出现警告信息,不用管,只要显示True,说明的CUDA配置成功了!!!!

http://www.lryc.cn/news/336491.html

相关文章:

  • STL之map
  • 闲谈2024(一)
  • SQL注入利用 学习- 布尔盲注
  • 前端项目部署教程——有域名有证书
  • 《看漫画学C++》第12章 可大可小的“容器”——向量
  • OpenAI推出GPTBot网络爬虫:提升AI模型同时引发道德法律争议
  • Claude使用教程
  • 【经典算法】LeetCode25:K 个一组翻转链表(Java/C/Python3,Hard)
  • 6.11物联网RK3399项目开发实录-驱动开发之定时器的使用(wulianjishu666)
  • 算法训练营第二十三天(二叉树完结)
  • mysql主从复制Slave_SQL_Running: No
  • 【SpringBoot】SpringBoot项目快速搭建
  • Terraform 状态不同步处理
  • 4.2.k8s的pod-标签管理、镜像拉取策略、容器重启策略、资源限制、优雅终止
  • 能源党建后台项目总结
  • 股票高胜率的交易法则是什么?
  • C语言 | sizeof与strlen的区别(附笔试题)
  • AI自动绘画器介绍和应用场景
  • java二叉树前中后序遍历
  • 【LeetCode刷题笔记】LeetCode 1365.有多少小于当前数字的数字
  • 室内定位中文综述阅读
  • 微信小程序uniapp+vue电力巡线任务故障报修管理系统2q91t
  • springboot国际化多语言
  • set和map
  • Open CASCADE学习|求曲面的参数空间
  • 代码随想录阅读笔记-二叉树【总结】
  • 【SpringBoot整合系列】SpringBoot整合FastDFS(二)
  • L2-2 巴音布鲁克永远的土(二分+并查集)
  • Spring Cloud学习笔记:Eureka简介,Eureka简单样例
  • 【漏洞复现】WordPress Welcart 任意文件读取漏洞(CVE-2022-4140)