- torch.cuda.is_available() cuda是否可用；
- torch.cuda.device_count() 返回gpu数量；
- torch.cuda.get_device_name(0) 返回gpu名字，设备索引默认从0开始；
- torch.cuda.current_device()；
- cuda是nvidia gpu的编程接口，opencl是amd gpu的编程接口
pytorch gpu
猜你喜欢
转载自blog.csdn.net/weixin_36670529/article/details/104030953
今日推荐
周排行