Convert the pytorch tensor to a numpy array: numpy_tensor ... 你可以调用torch.cuda.is_available() 函数,检查PyTorch 中是否有可用CUDA。
確定! 回上一頁