一元网络论坛

 找回密码
 立即注册
搜索
热搜: 活动 交友 discuz
查看: 124|回复: 0

如何顺利安装 PyTorch 依赖库?

[复制链接]

3万

主题

3万

帖子

9万

积分

管理员

Rank: 9Rank: 9Rank: 9

积分
96158
发表于 2024-9-29 18:24:23 | 显示全部楼层 |阅读模式
Torch拥有多个版本,包括官网提供的CPU、GPU(CUDA多个版本、ROCm)等,以及Windows上的DirectML加速库 `torch_directml`、Intel的 `intel_extension_for_pytorch` 等。  这些版本繁多,是否存在一个一键包,自动判断版本并安装对应依赖? 希望能得到各位大佬的推荐。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Archiver|手机版|小黑屋|一元网络论坛

GMT+8, 2024-11-26 11:30 , Processed in 0.072356 second(s), 19 queries .

Powered by Discuz! X3.4

Copyright © 2001-2020, Tencent Cloud.

快速回复 返回顶部 返回列表