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

相关推荐
时见先生16 小时前
Python库和conda搭建虚拟环境
开发语言·人工智能·python·自然语言处理·conda
昨夜见军贴061618 小时前
IACheck AI审核在生产型企业质量控制记录中的实践探索——全面赋能有关物质研究合规升级
大数据·人工智能
智星云算力18 小时前
智星云镜像共享全流程指南,附避坑手册(新手必看)
人工智能
盖雅工场18 小时前
驱动千店销售转化提升10%:3C零售门店的人效优化实战方案
大数据·人工智能·零售·数字化管理·智能排班·零售排班
Loo国昌18 小时前
深入理解 FastAPI:Python高性能API框架的完整指南
开发语言·人工智能·后端·python·langchain·fastapi
发哥来了18 小时前
【AI视频创作】【评测】【核心能力与成本效益】
大数据·人工智能
醉舞经阁半卷书119 小时前
Python机器学习常用库快速精通
人工智能·python·深度学习·机器学习·数据挖掘·数据分析·scikit-learn
产品何同学20 小时前
在线问诊医疗APP如何设计?2套原型拆解与AI生成原型图实战
人工智能·产品经理·健康医疗·在线问诊·app原型·ai生成原型图·医疗app
星爷AG I20 小时前
9-14 知觉整合(AGI基础理论)
人工智能·agi
开源技术20 小时前
Violit: Streamlit杀手,无需全局刷新,构建AI面板
人工智能·python