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

相关推荐
卷福同学9 小时前
不用服务器,不用配环境,我10分钟上线了一个AI Agent
人工智能·后端·算法
码兄科技10 小时前
Java AI智能体开发实战:从零构建智能对话系统指南
java·开发语言·人工智能
Zik----10 小时前
CCswitch-code
人工智能
AI科技星11 小时前
全域谱分析:无穷维超复数信息场分形统一场论 ——自然、量子、金融多重分形第一性原理完整体系(中英双语终稿)
人工智能·机器学习·金融·乖乖数学·全域数学
stormzhangV12 小时前
为什么你的 AI 像智障
人工智能
ai产品老杨12 小时前
H264 H265视频分析常见问题和排查清单
人工智能·算法·音视频
项目经理老王12 小时前
OpenClaw无捆绑安装包,安全纯净版AI助手部署
人工智能·安全
梦帮科技13 小时前
GRAVIS v4.0:基于Web的极速套利架构设计与实时数据流实现
前端·人工智能·rust·自动化·区块链·智能合约·数字货币
“码”力全开13 小时前
AI视频分析API性能优化指南
人工智能·性能优化·音视频
liuyicenysabel13 小时前
大模型学习笔记 · 第八篇 · 进阶:偏好对齐与多卡训练
人工智能·笔记·学习