检验 pytorch是否可以使用GPU Cuda，以及Cuda版本

原创于 2020-11-28 09:49:27 发布 · 3.4k 阅读

13 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #python #pycharm

pytorch相关专栏收录该内容

25 篇文章

订阅专栏

本文介绍了如何使用PyTorch并检查其版本信息，验证CUDA是否可用及其版本，获取GPU的数量、名称及当前设备索引等内容。

部署运行你感兴趣的模型镜像

import torch

# torch版本
print(" pytorch_version{}".format(torch.__version__))

# cuda是否可用、以及版本
print("\nCUDA is avaiable:{}, version is {}".
        format(torch.cuda.is_available(), torch.version.cuda))


# 返回gpu数量
print(torch.cuda.device_count())

# 返回gpu名字，设备索引默认从0开始
print(torch.cuda.get_device_name(0))

# 返回当前设备索引
print(torch.cuda.current_device())

您可能感兴趣的与本文相关的镜像