查询pytorch和cuda版本
时间: 2023-09-22 21:09:35 浏览: 103
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
你可以通过以下代码查询PyTorch和CUDA的版本信息:
```python
import torch
print("PyTorch version:", torch.__version__)
print("CUDA version:", torch.version.cuda)
```
运行这段代码,你将获得PyTorch和CUDA的版本信息。请注意,CUDA版本仅在使用GPU时才会显示。
阅读全文