【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/

接着测试

py 复制代码
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

相关推荐
斐夷所非5 分钟前
人工智能 AI. 机器学习 ML. 深度学习 DL. 神经网络 NN 的区别与联系
人工智能
递归不收敛19 分钟前
吴恩达机器学习课程(PyTorch适配)学习笔记:2.4 激活函数与多类别处理
pytorch·学习·机器学习
F_D_Z1 小时前
数据集相关类代码回顾理解 | StratifiedShuffleSplit\transforms.ToTensor\Counter
python·torchvision·transforms
Funny_AI_LAB2 小时前
OpenAI DevDay 2025:ChatGPT 进化为平台,开启 AI 应用新纪元
人工智能·ai·语言模型·chatgpt
深瞳智检2 小时前
YOLO算法原理详解系列 第002期-YOLOv2 算法原理详解
人工智能·算法·yolo·目标检测·计算机视觉·目标跟踪
tao3556672 小时前
【Python刷力扣hot100】283. Move Zeroes
开发语言·python·leetcode
深眸财经2 小时前
机器人再冲港交所,优艾智合能否破行业困局?
人工智能·机器人
小宁爱Python3 小时前
从零搭建 RAG 智能问答系统1:基于 LlamaIndex 与 Chainlit实现最简单的聊天助手
人工智能·后端·python
湖南人爱科技有限公司3 小时前
RaPhp和Python某音最新bd-ticket-guard-client-data加密算法解析(视频评论)
android·python·php·音视频·爬山算法·raphp
新知图书3 小时前
Encoder-Decoder架构的模型简介
人工智能·架构·ai agent·智能体·大模型应用开发·大模型应用