pytorch.version.cuda
时间: 2023-08-13 09:17:38 浏览: 131
pytorch中使用cuda扩展的实现示例
`torch.version.cuda` 是 PyTorch 模块中的一个属性,用于获取 PyTorch 编译时使用的 CUDA 版本。如果您已经安装了 PyTorch 并且已经成功启用了 CUDA,您可以在 Python 解释器中输入以下代码来获取 PyTorch 编译时使用的 CUDA 版本号:
```python
import torch
print(torch.version.cuda)
```
执行这个代码会输出当前 PyTorch 编译时使用的 CUDA 版本号,例如:
```
10.2
```
这表示 PyTorch 编译时使用的是 CUDA 10.2 版本。
阅读全文