开发者社区 问答 正文

调用pytorch函数的代码是什么?

调用pytorch函数的代码是什么?

展开
收起
游客mm5dulkpivqdo 2021-12-02 23:12:05 384 分享 版权
1 条回答
写回答
取消 提交回答
  • import torch torch.cuda.set_device(id)

    2021-12-02 23:12:28
    赞同 展开评论
问答分类:
问答标签:
问答地址: