5060显卡Windows配置Anaconda中的CUDA及Pytorch

1. 在Anaconda中创建相关虚拟环境

注:Torch2.8需要python版本大于等于3.9

我这里直接3.10版本创建(py310cuda128torch28是我起的环境名称,但其实torch装了2.7版本)

conda create --name py310cuda128torch28 python=3.10

2.Pytorch安装

直接官网上最高版本

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128

放心cuda版本自己就安装好了

5. 试验一下显卡可不可以用

import torch

print(f"PyTorch 版本: {torch.version}")

print(f"CUDA 是否可用: {torch.cuda.is_available()}")

print(f"CUDA 版本: {torch.version.cuda}")

print(f"GPU 设备: {torch.cuda.get_device_name(0)}")

ok完成

相关推荐
心念枕惊5 小时前
零基础认识大语言模型(LLM)工作原理(9.从聊天机器人到智能体:AI 为什么必须学会完成任务?)
人工智能·语言模型·机器人
IT_陈寒5 小时前
Python的finally居然不等同于Go的defer,差点坑惨我
前端·人工智能·后端
码农学院5 小时前
GEO效果度量体系:AI搜索可见性、引用率与转化追踪的技术实现
人工智能·dubbo
恋猫de小郭5 小时前
给 AI 的 Agent 实现指南,可控 Agent 的关键
前端·人工智能·ai编程
AI导出鸭PC端5 小时前
手机deepseek怎么导出pdf AI导出鸭
人工智能·pdf
蓦然回首却已人去楼空5 小时前
Build a Large Language Model (From Scratch) 第6章 针对分类的微调
人工智能·语言模型·分类
核数聚5 小时前
具身智能数据采集:从实验室到生活,核数聚如何为机器人 “喂饱” 真实数据
人工智能·机器人·生活·核数聚
qq_454245035 小时前
rules prompt
人工智能·prompt
夏贰四5 小时前
数据资产平台如何落地企业数据分级合规管控?数据资产平台怎样满足行业数据监管审查要求?
大数据·数据库·人工智能
工业机器视觉设计和实现5 小时前
动量的好处与困扰(二,摸到pytorch尾灯!)
人工智能·pytorch·cudnn微积分