- 1、cuda是否可用
- 2、cuda的gpu数量
- 3、cuda的gpu名称
- 4、cuda当前的gpu索引
接口:torch.cuda.is_available()
如下示例,cuda可用:
root@3dd53ab5928d:/# python3 Python 3.6.9 (default, Jan 26 2021, 15:

接口:torch.cuda.is_available()
如下示例,cuda可用:
root@3dd53ab5928d:/# python3 Python 3.6.9 (default, Jan 26 2021, 15: