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版本
相关推荐
企企通采购云平台2 分钟前
「天元宠物」×企企通,加速数智化升级,“链”接萌宠消费新蓝海
大数据·人工智能·宠物
超级小忍6 分钟前
Spring AI ETL Pipeline使用指南
人工智能·spring
张较瘦_29 分钟前
[论文阅读] 人工智能 | 读懂Meta-Fair:让LLM摆脱偏见的自动化测试新方法
论文阅读·人工智能
巴伦是只猫1 小时前
【机器学习笔记 Ⅲ】4 特征选择
人工智能·笔记·机器学习
好心的小明1 小时前
【王树森推荐系统】召回11:地理位置召回、作者召回、缓存召回
人工智能·缓存·推荐系统·推荐算法
weixin_418813872 小时前
Python-可视化学习笔记
笔记·python·学习
lishaoan772 小时前
使用tensorflow的线性回归的例子(十二)
人工智能·tensorflow·线性回归·戴明回归
Danceful_YJ2 小时前
4.权重衰减(weight decay)
python·深度学习·机器学习
二DUAN帝2 小时前
UE实现路径回放、自动驾驶功能简记
人工智能·websocket·机器学习·ue5·自动驾驶·ue4·cesiumforue
Zonda要好好学习3 小时前
Python入门Day5
python