【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

相关推荐
xiangduanjava几秒前
关于安装Ollama大语言模型本地部署工具
人工智能·语言模型·自然语言处理
zzywxc78721 分钟前
AI 正在深度重构软件开发的底层逻辑和全生命周期,从技术演进、流程重构和未来趋势三个维度进行系统性分析
java·大数据·开发语言·人工智能·spring
超龄超能程序猿23 分钟前
(1)机器学习小白入门 YOLOv:从概念到实践
人工智能·机器学习
大熊背33 分钟前
图像处理专业书籍以及网络资源总结
人工智能·算法·microsoft
3gying38 分钟前
chromedriver
python
江理不变情39 分钟前
图像质量对比感悟
c++·人工智能
DES 仿真实践家2 小时前
【Day 11-N22】Python类(3)——Python的继承性、多继承、方法重写
开发语言·笔记·python
张较瘦_3 小时前
[论文阅读] 人工智能 + 软件工程 | 需求获取访谈中LLM生成跟进问题研究:来龙去脉与创新突破
论文阅读·人工智能
一 铭4 小时前
AI领域新趋势:从提示(Prompt)工程到上下文(Context)工程
人工智能·语言模型·大模型·llm·prompt
云泽野6 小时前
【Java|集合类】list遍历的6种方式
java·python·list