pytorch 查看cuda版本

您可以使用以下代码来查看您的 PyTorch 是否支持 CUDA，以及您的 CUDA 版本：

import torch

print(torch.version.cuda)  # 打印 CUDA 版本
print(torch.cuda.is_available())  # 检查是否支持 CUDA

如果您的 PyTorch 不支持 CUDA，或者您没有安装 CUDA，那么 `torch.cuda.is_available()` 将返回 `False`。

相关问题

查看pytorch版本cuda版本

你可以使用以下代码来查看PyTorch和CUDA的版本：

import torch

print("PyTorch version:", torch.__version__)
print("CUDA version:", torch.version.cuda)

运行这段代码将会打印出你当前安装的PyTorch和CUDA的版本信息。

查看pytorch的cuda版本

您可以使用以下代码查看您的PyTorch是否支持CUDA以及CUDA的版本：

import torch
if torch.cuda.is_available():
    print('CUDA version:', torch.version.cuda)
else:
    print('CUDA is not available')