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完成

相关推荐
恋猫de小郭21 小时前
Redis 作者反驳「中国模型之所以强,是因为通过 API 蒸馏了美国模型」
前端·人工智能·ai编程
林间码客21 小时前
04 ROC曲线与AUC:从零开始手动计算
大数据·人工智能·算法
codexu21 小时前
NoteGen 里一条记录如何变成 Markdown
人工智能
澄旭21 小时前
拆解一个成熟 Skill,看懂 Skill 到底该怎么写
人工智能
王木风21 小时前
拆解一个 LLM 工程化项目:16 个 Service + Agent 对话循环怎么协同跑流水线
人工智能·agent
沪漂阿龙21 小时前
《LangChain 系列》Human-in-the-loop:什么时候必须让人工介入?
人工智能·架构·langchain
冬哥聊AI21 小时前
Loop Engineering 来了:从写 Prompt 到设计 Loop,AI 编程的第四次范式跃迁
人工智能
柒星栈21 小时前
Codex 不只是更强的代码助手,它开始像代理一样推进开发任务了
人工智能
o_insist21 小时前
04-从零手写 ReAct 循环:Agent 的心跳是怎么转起来的
人工智能·agent
DayByDay21 小时前
从“单专家”到“多专家辩论”:多大脑对话实现复盘
人工智能