2021年3月1日
Pytorch显卡设置
摘要： 参考资料： https://zhuanlan.zhihu.com/p/166161217 本人最常使用到显卡和CUDA的东西莫过于Pytorch了。这篇文章着重说明两个问题：1. 如何import torch并使之输出比较完备的CUDA信息 2. 在服务器上有多张卡的环境下，如何使任务在特定的卡或特 阅读全文
posted @ 2021-03-01 13:29 思念殇千寻 阅读(1032) 评论(0) 推荐(0) 编辑