2021年5月31日
修复torch.cuda.is_available()返回False问题
摘要： 学习官方60分钟教程时，尝试把张量移到GPU tensor = torch.rand(3, 4) # We move our tensor to the GPU if available if torch.cuda.is_available(): tensor = tensor.to('cuda') 阅读全文
posted @ 2021-05-31 23:19 奥巴荣 阅读(8849) 评论(0) 推荐(0) 编辑