win11用一条命令给anaconda环境安装GPU版本pytorch,并检查是否为GPU版本

不需要安装cudatookit和cudnn,只要有显卡就有cuda,安装tenserflow-gpu的时候需要额外安装的是cudatookit和cudnn


用一条命令即可

打开官网:

Start Locally | PyTorch

选择安装版本

运行命令即可

复制代码
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

pytorch默认安装的就是GPU的,如果需要安装CPU的,要加一句cpuonly,参考下面的链接:
Previous PyTorch Versions | PyTorch


如何查看自己的pytorch是gpu还是cpu?

输入:

复制代码
>>> import  torch 
>>> torch.cuda.is_available()
 若返回为True,则使用的是GPU版本的torch,若为False,则为CPU版本
相关推荐
七牛云行业应用几秒前
GPT-5.6 三款模型核心差距:Sol、Terra、Luna 怎么选
人工智能·gpt
Hotchip_MEMS19 分钟前
智能手表与语音交互:MP381A-AB08E MEMS麦克风的低功耗方案
人工智能·物联网·制造
chenyulin454520 分钟前
企业微信API二次开发:万级并发回调下的极致幂等性设计与防重放架构实战
大数据·人工智能·安全·架构·企业微信
浩哥学JavaAI34 分钟前
2026年最新AI agent面试(06)_RAG文档与检索
人工智能·面试·职场和发展
爱吃苹果的梨叔36 分钟前
2026年指挥中心分布式坐席怎么选?清虹创智让多系统、多坐席和大屏真正协同
python
Adgine梦想家43 分钟前
RAG三步骤图解:让大模型实时理解你的网站
人工智能
我是小杰啊1 小时前
幻视是什么?一款用自然语言搜索图片与视频的 AI 视觉问答工具
人工智能·计算机视觉·多模态·ai应用·视频搜索
就是一顿骚操作1 小时前
神经网络可解释性积木:从特征可视化到归因地图
人工智能·深度学习·神经网络
kisshyshy1 小时前
从无崖子到OpenAI:大模型间的“传功”,动了谁的奶酪?
人工智能·深度学习·设计模式
起予者汝也1 小时前
Python 数据结构
开发语言·数据结构·python