1. 是否可以用显卡：
torch.cuda.is_available()
2. 显卡个数：
torch.cuda.device_count()
3. 显卡名字：
torch.cuda.get_device_name(0)
黄世宇/Shiyu Huang's Personal Page：https://huangshiyu13.github.io/
Pytorch检查cuda
阅读 69
2022-07-18
1. 是否可以用显卡：
torch.cuda.is_available()
2. 显卡个数：
torch.cuda.device_count()
3. 显卡名字：
torch.cuda.get_device_name(0)
黄世宇/Shiyu Huang's Personal Page：https://huangshiyu13.github.io/
相关推荐
精彩评论（0）
