设为首页收藏本站

开启辅助访问切换到窄版

Hello Mat»论坛 › C++/MFC › 进阶使用 › libtorch选择指定GPU

发新帖

查看: 1114|回复: 0

libtorch选择指定GPU

1315 主题	1541 帖子	0 金钱

Rank: 9 Rank: 9 Rank: 9

积分: 22607

发表于 2024-8-16 15:11:21 来自手机 | 显示全部楼层 |阅读模式

http://t.csdnimg.cn/ZL8JV

torch::jit::script::Module module = torch::jit::load(filename,torch::Device(torch::DeviceType::CUDA,gpu_id));//加载模型

复制代码

torch:: Device(torch:: DeviceType::CUDA,gpu_id)这个别省略了

回复

使用道具举报

发新帖

Python|Opencv|MATLAB|Halcom.cn ( 蜀ICP备16027072号 )

GMT+8, 2024-10-18 16:34 , Processed in 0.216395 second(s), 21 queries .

Powered by Discuz! X3.4

Copyright © 2001-2021, Tencent Cloud.

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