python安装torch-cluster、torch-scatter、torch-sparse和torch-geometric | torch_geometric详细安装教程

最新推荐文章于 2025-05-26 15:33:11 发布

原创

最新推荐文章于 2025-05-26 15:33:11 发布 · 5k 阅读

51 ·

CC 4.0 BY-SA版权

文章标签：

#python #开发语言

1.检查CUDA版本

【方法1】用nvidia-smi

已装cuda，用nvidia-smi 或 nvcc -V 查看
在这里插入图片描述

【方法2】用torch

已装torch，用torch代码打印

import torch
print(torch.__version__)                # 查看pytorch安装的版本号
print(torch.cuda.is_available())        # 查看cuda是否可用。True为可用，即是gpu版本pytorch
print(torch.cuda.get_device_name(0))    # 返回GPU型号
print(torch.cuda.device_count())        # 返回可以用的cuda（GPU）数量，0代表一个
print(torch.version