linux服务器查看pytorch和cuda的版本
时间: 2023-12-06 08:03:43 浏览: 162
pytorch 查看cuda 版本方式
要查看服务器上安装的PyTorch和CUDA版本,可以按照以下步骤进行:
1. 查看CUDA版本:
在终端中输入以下命令查看CUDA版本:
```bash
nvcc --version
```
该命令将显示CUDA的版本号。
2. 查看PyTorch版本:
在Python交互式命令行或Python脚本中,输入以下命令来查看PyTorch版本:
```python
import torch
print(torch.__version__)
```
该命令将打印出安装的PyTorch版本号。
如果服务器上未安装PyTorch或CUDA,则可以按照官方文档进行安装。
阅读全文