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版本
相关推荐
落雪财神意40 分钟前
股指10月想法
大数据·人工智能·金融·区块链·期股
中杯可乐多加冰40 分钟前
无代码开发实践|基于业务流能力快速开发市场监管系统,实现投诉处理快速响应
人工智能·低代码
渣渣盟42 分钟前
解密NLP:从入门到精通
人工智能·python·nlp
新智元43 分钟前
万亿级思考模型,蚂蚁首次开源!20 万亿 token 搅局开源 AI
人工智能·openai
老程序员刘飞44 分钟前
注册 区块链节点
python
算家计算1 小时前
阿里开源最强视觉模型家族轻量版:仅4B/8B参数,性能逼近72B旗舰版
人工智能·开源·资讯
KAIWEILIUCC1 小时前
Python抽象基类(abc.ABC)介绍
python
MarkHD1 小时前
Dify从入门到精通 第16天 工作流进阶 - 分支与判断:构建智能路由客服机器人
人工智能·机器人
意疏1 小时前
从告警风暴到根因定位:SigNoz+CPolar让分布式系统观测效率提升10倍的实战指南
人工智能
独行soc1 小时前
2025年渗透测试面试题总结-106(题目+回答)
网络·python·安全·web安全·adb·渗透测试·安全狮