查看torch和cuda是否安装后可用

饿了就干饭

于 2024-06-17 09:16:29 发布

阅读量346

点赞数 2

分类专栏： Pytorch 文章标签： torch

本文链接：https://blog.csdn.net/qq_45893319/article/details/139731789

版权

Pytorch 专栏收录该内容

22 篇文章 1 订阅

订阅专栏

import torch

# 检查CUDA是否可用
cuda_available = torch.cuda.is_available()
if cuda_available:
    print(f"CUDA is available with version: {torch.version.cuda}")
else:
    print("CUDA is not available.")

# 检查PyTorch版本
print(f"PyTorch version: {torch.__version__}")

# 检查CUDA的device属性
if cuda_available:
    device = torch.device("cuda")
    print(f"CUDA device name: {torch.cuda.get_device_name(0)}")
    print(f"CUDA device memory: {torch.cuda.get_device_properties(0).total_memory / 1024 ** 2} MB")

输出：

"D:\software install location\anaconda\envs\idm\python.exe" "E:/008 virtual_clothes/IDM-VTON/测试.py"
CUDA is available with version: 11.8
PyTorch version: 2.0.1+cu118
CUDA device name: NVIDIA GeForce RTX 4080
CUDA device memory: 16375.375 MB

进程已结束，退出代码为 0