torch.cuda. 该包增加了对CUDA张量类型的支持,实现了与CPU张量相同的功能,但使用GPU进行计算。 它是懒惰的初始化,所以你可以随时导入它,并使用 is_available() 来 ...
確定! 回上一頁