怎样查看安装的pytorch+cuda版本
时间: 2024-03-24 15:34:02 浏览: 68
pytorch 查看cuda 版本方式
5星 · 资源好评率100%
要查看已安装的PyTorch和CUDA版本,可以使用以下代码:
```python
import torch
print("PyTorch版本:", torch.__version__)
print("CUDA是否可用:", torch.cuda.is_available())
if torch.cuda.is_available():
print("CUDA版本:", torch.version.cuda)
```
这段代码首先导入torch模块,然后使用`torch.__version__`打印PyTorch的版本号。接下来,使用`torch.cuda.is_available()`检查CUDA是否可用,并打印结果。如果CUDA可用,可以使用`torch.version.cuda`打印CUDA的版本号。
阅读全文