分类：判断pytorch

2021
07-26

如何判断pytorch是否支持GPU加速

加上这句代码：printtorch.cuda.is_available()判断完毕！说说在pytorch中如何查看gpu信息吧~为什么将数据转移至GPU的方法叫做.cuda而不是.gpu，就像将数据转移至CPU调用的方法是.cpu？这是因为GPU的编程接口采用CUDA，而目前并不是所有的GPU都支持CUDA，只有部分Nvidia的GPU才支持。PyTorch未来可能会支持AMD的GPU，而AMDGPU的编程接口采用OpenCL，因此PyTorch还预留着.cl方法，用于以后支持AMD等的GPU。torch.cuda.is_availabl... 继续阅读 >