查看显卡算力和cuda

以下是查看显卡算力和CUDA版本的方法：

1. 查看显卡算力

可以通过以下代码来查看显卡算力：

import torch
if torch.cuda.is_available():
    print(torch.cuda.get_device_name(0))
    print('Compute Capability:', torch.cuda.get_device_capability(0))
else:
    print('No GPU available')

其中，`torch.cuda.get_device_name(0)`可以获取第一个GPU的名称，`torch.cuda.get_device_capability(0)`可以获取第一个GPU的算力。

2. 查看CUDA版本

可以通过以下命令来查看CUDA版本：

nvcc --version

如果你想在Python中查看CUDA版本，可以使用以下代码：

import torch
print(torch.version.cuda)

如果你想查看当前显卡驱动支持的最高CUDA版本，可以使用以下命令：

nvidia-smi

在输出中，你可以看到当前显卡驱动版本和支持的最高CUDA版本。