概述
为什么将数据转移至GPU的方法叫做.cuda而不是.gpu，就像将数据转移至CPU调用的方法是.cpu？这是因为GPU的编程接口采用CUDA，而目前并不是所有的GPU都支持CUDA，只有部分Nvidia的GPU才支持。PyTorch未来可能会支持AMD的GPU，而AMD GPU的编程接口采用OpenCL，因此PyTorch还预留着.cl方法，用于以后支持AMD等的GPU。
-
torch.cuda.is_available()
cuda是否可用； -
torch.cuda.device_count()
返回gpu数量； -
torch.cuda.get_device_name(0)
返回gpu名字，设备索引默认从0开始； -
torch.cuda.current_device()
返回当前设备索引；
更多信息：https://pytorch.org/docs/stable/cuda.html
最后
以上就是繁荣中心为你收集整理的pytorch中查看gpu信息的全部内容，希望文章能够帮你解决pytorch中查看gpu信息所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
![[实时渲染] 3.1 GPU管线综述](/uploads/reation/bcimg13.png)
发表评论 取消回复