查看gpu版本pytorch版本号
时间: 2023-08-17 13:04:34 浏览: 164
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
您可以使用以下代码来查看GPU版本和PyTorch版本号:
```python
import torch
# 检查GPU是否可用
if torch.cuda.is_available():
print("GPU版本:", torch.version.cuda)
else:
print("没有可用的GPU")
# 打印PyTorch版本号
print("PyTorch版本号:", torch.__version__)
```
运行这段代码后,您将看到输出结果中包含GPU版本和PyTorch版本号。如果您的系统中没有可用的GPU,将会显示"没有可用的GPU"。
阅读全文