print(torch.cuda.is_available) <function is_available at 0x2b36287e70a0>
时间: 2023-12-16 07:25:34 浏览: 200
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
`print(torch.cuda.is_available)`是一个函数,用于检查当前系统是否支持CUDA(Compute Unified Device Architecture,种并行计算平台和编程模型)。返回True,则表示CUDA可用,否则表示不可用。在深度学习中,CUDA通常用于加速神经网络的训练和推理过程。
阅读全文