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

【CUDA pytorch】

ev= win10
3050ti
联想笔记本

nvcc --version

得到

PS C:\Users\25515> nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Tue_May__3_19:00:59_Pacific_Daylight_Time_2022
Cuda compilation tools, release 11.7, V11.7.64
Build cuda_11.7.r11.7/compiler.31294372_0

CUDA版本是 11.7

PS C:\Users\25515> wmic path win32_videocontroller get name
Name
NVIDIA GeForce RTX 3050 Ti Laptop GPU
Intel(R) Iris(R) Xe Graphics

下载torch

https://pytorch.org/get-started/locally/

在这里插入图片描述

接着测试


import torch# 创建张量
x = torch.tensor([1, 2, 3])
y = torch.tensor([4, 5, 6])# 张量加法
z = x + y
print(z)  # 输出 tensor([5, 7, 9])# 测试 GPU 张量(如果 CUDA 可用)
if torch.cuda.is_available():x_gpu = x.to("cuda")print(x_gpu)  # 输出类似 tensor([1, 2, 3], device='cuda:0')
else:print("CUDA 不可用")pass
print(torch.__version__)

得到
在这里插入图片描述

有个大兄弟在卖代码,等我搓搓,哈哈

https://gitee.com/mqwdasddqw/project-download-address

换源

在这里插入图片描述
在这里插入图片描述

项目结构

handwritten_digit_recognition/

├── a3_cnn.py # 包含 CNN 模型定义
├── model/
│ └── mnist_model.pkl # 保存的模型文件
└── main.py # 主程序文件(以下为完整代码)

hand_writing_number

http://www.dtcms.com/a/166599.html

相关文章:

  • 基于连接感知的实时困倦分类图神经网络
  • kibana重建es索引
  • ShardingSphere5详细笔记
  • OpenCV-Python (官方)中文教程(部分一)_Day20
  • 软件架构选型之“如何选”
  • 生物化学笔记:神经生物学概论05 感受野 视觉中枢 高级视皮层中的信息走向
  • 【python实用小脚本-43】用Python自动发送生日祝福,让情感更高效
  • 希尔伯特第十问题:是一个伪命题
  • 时态--00--总述
  • 代码随想录打卡|Day31动态规划(最后一块石头的重量2、目标和、一和零)
  • 如何在 Linux 环境下使用 Certbot 自动生成 SSL 证书并部署到 Nginx 服务中
  • POSIX介绍
  • SD - WAN 跨境网络专线部署方式介绍
  • 国产工业软件突破路径
  • 电脑干货:开源免费的QQ空间说说下载工具GetQzonehistory
  • 机器指标监控技术方案
  • FreeBSD升级到14.2后启动的时候提示BOOT LOADER IS TOO OLD. PLEASE UPGRADE.
  • 利用v0与Cursor优化开发流程和效率
  • 六.割草机技术总结--6.RTK定位精度分析
  • Arduino程序结构详解与嵌入式开发对比指南
  • 7.软考高项(信息系统项目管理师)-资源管理
  • 中国1km分辨率1901-2023年均气温降水数据
  • DotNet 入门:(二) 项目运行
  • 分组密码算法ShengLooog设计原理详解
  • Kubernetes 配置中的 Selector 详解
  • 验证加密与数字签名实验
  • 英语四级听力疑难知识点详解
  • WebGL图形编程实战【5】:层次构建 × Shader初始化深度剖析
  • 可视化网页自动化流程管理工具
  • 实验-单总线温度采集与显示(汇编语言与接口技术)