2022年7月11日
Pytorch检查cuda
摘要： 1. 是否可以用显卡： torch.cuda.is_available() 2. 显卡个数： torch.cuda.device_count() 3. 显卡名字： torch.cuda.get_device_name(0) 设置可见GPU: CUDA_VISIBLE_DEVICES=1 python 阅读全文
posted @ 2022-07-11 17:32 Shiyu_Huang 阅读(442) 评论(0) 推荐(0) 编辑