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版本
相关推荐
第九序言T3 分钟前
Python 数据处理进阶(十八)批量处理同结构文件:一个文件夹一次跑完
开发语言·python·自动化·pandas·数据处理
Shulex8 分钟前
跨境电商 AI Agent 核心架构解析:从只读问答到自主执行的工程落地
人工智能·架构·智能客服·跨境电商
Java后端的Ai之路10 分钟前
CAS并发控制-从CPU指令到企业级乐观锁
vue.js·人工智能·python·cas·乐观锁
qq_4260039614 分钟前
UI自动化高频次登录login限流解决方案
python·ui·自动化
情 九15 分钟前
MCP 实战 06:把 HTTP 接口接成工具,让 AI 帮你查数据
人工智能·网络协议·http·mcp
Sakura_j_h_33 分钟前
EasyEDA PCB 快速设计Skills
c++·python·github
码农小旋风1 小时前
Ray分布式计算引擎介绍
人工智能·架构
火山引擎和TA的超级拍档们1 小时前
产品情报局 08 | AI 内容助手:全网热点驱动,一站式打通内容生产全链路
人工智能
Binovate缤江数智1 小时前
水泵远程开关之后,还需要管什么?数智开关在无人值守泵站中的监测、保护与能耗分析
人工智能·物联网