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版本
相关推荐
马***4114 小时前
适配成人英语学习痛点,打造落地性强的学习辅助方式
人工智能·学习
夜焱辰4 小时前
浏览器端 Agent 的文件版本管理:不用 Git,基于 OPFS + SQLite 自己造了一个
前端·人工智能
Ricky05534 小时前
CTRL-WORLD:一种用于机器人操控的可控生成世界模型(中美2025年联合研究)
人工智能·机器人·世界模型
jeffer_liu4 小时前
Spring AI 生产级实战:工具调用
java·人工智能·后端·spring·ai编程
lifloveyou4 小时前
table接口结构
python
阿乔外贸日记4 小时前
2026尼日利亚五项清关政策更新,拉高能源装备进口综合成本
大数据·人工智能·搜索引擎·智能手机·云计算·能源
民乐团扒谱机4 小时前
【AI笔记】短时纯音时长对音高感知偏移效应研究综述
人工智能·笔记
侃谈科技圈5 小时前
破除数据中台落地困境:2026数据治理平台差异化能力与选型决策指南
大数据·人工智能
大象说5 小时前
Python多进程共享队列无报错僵死 120G Nginx访问日志清洗踩坑全记录
人工智能·自然语言处理
Cosolar5 小时前
AutoGen 精通教程:从零到企业级多 Agent 系统架构师
人工智能·后端·面试