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版本
相关推荐
用户27784491049937 分钟前
Python打造Excel记账模板,摸鱼时间也能轻松理财
人工智能·python
阿杜杜不是阿木木8 分钟前
03.使用spring-ai玩转MCP
java·人工智能·spring boot·spring·mcp·spring-ai
IT古董8 分钟前
【漫话机器学习系列】230.模型复杂度对训练和测试误差的影响(The Effect Of Model Complexity Training And Test)
人工智能
奕川10 分钟前
深入浅出:MCP 协议及应用
人工智能·后端
IT古董10 分钟前
【漫话机器学习系列】233.激活阈(Threshold Activation)
人工智能
极客智谷12 分钟前
MCP协议与智能体开发:构建动态协作AI系统的基石
人工智能
极客智谷12 分钟前
Spring AI应用系列——基于DeepSeek客户端的Chat应用
人工智能·后端
闲人编程14 分钟前
OpenCV图像矩与形状匹配完全指南
python·opencv·图像识别
带娃的IT创业者20 分钟前
《AI大模型应知应会100篇》第37篇:Agent框架入门:让AI具备自主行动能力
人工智能
OJAC近屿智能37 分钟前
新增29个专业,科技成为未来主赛道!
人工智能·科技·ai·机器人·aigc·近屿智能