【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

相关推荐
静心问道2 分钟前
TrOCR: 基于Transformer的光学字符识别方法,使用预训练模型
人工智能·深度学习·transformer·多模态
说私域4 分钟前
基于开源AI大模型、AI智能名片与S2B2C商城小程序源码的用户价值引导与核心用户沉淀策略研究
人工智能·开源
亲持红叶5 分钟前
GLU 变种:ReGLU 、 GEGLU 、 SwiGLU
人工智能·深度学习·神经网络·激活函数
说私域5 分钟前
线上协同办公时代:以开源AI大模型等工具培养网感,拥抱职业变革
人工智能·开源
群联云防护小杜7 分钟前
深度隐匿源IP:高防+群联AI云防护防绕过实战
运维·服务器·前端·网络·人工智能·网络协议·tcp/ip
惜.己10 分钟前
使用python读取json数据,简单的处理成元组数组
开发语言·python·测试工具·json
摘星编程12 分钟前
构建智能客服Agent:从需求分析到生产部署
人工智能·需求分析·智能客服·agent开发·生产部署
不爱学习的YY酱15 分钟前
信息检索革命:Perplexica+cpolar打造你的专属智能搜索中枢
人工智能
whaosoft-1431 小时前
51c自动驾驶~合集7
人工智能
都叫我大帅哥2 小时前
Python的Optional:让你的代码优雅处理“空值”危机
python