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

相关推荐
Dfreedom.2 分钟前
集成学习完全解析:从核心思想到常见误区
人工智能·机器学习·集成学习
杜子不疼.3 分钟前
Linux 部署 RocketMQ 实操:从内网到公网,搞定远程消息服务
linux·运维·人工智能·rocketmq
新缸中之脑5 分钟前
NotebookLM:最佳学习工具
人工智能·学习·chatgpt
8Qi85 分钟前
Hello-Agents阅读笔记--Reflection
人工智能·llm·agent·智能体
LightYoungLee5 分钟前
大模型(七)Agent AI学习笔记
人工智能·笔记·学习
Men-DD5 分钟前
AI(四): 工具 (GitNexus、CLI-Anything)
人工智能·ai·gitnexus·cli-anything
smilejingwei6 分钟前
从同质化市场搅局的开源BI 到AI时代全链智胜的ChatBI
人工智能·bi·chatbi
avi91117 分钟前
【AI相关】-Gradio和 Streamlit,怎么选, Streamlit 篇
人工智能·aigc·gradio·streamlit
Coovally AI模型快速验证7 分钟前
2.5GB 塞进浏览器:Mistral 开源实时语音识别,延迟不到半秒
人工智能·5g·计算机视觉·开源·语音识别
AI营销资讯站8 分钟前
AI营销内容增长瓶颈?原圈科技以AI Agents破局之道
大数据·人工智能