ubuntu查看torch和cuda版本

在Ubuntu系统中，你可以通过命令行来查看已安装的PyTorch和CUDA版本。首先，你需要检查是否有安装CUDA：

1. 打开终端：

   sudo apt update && sudo apt upgrade

2. 安装`nvidia-smi`工具（如果你还没有安装，用于管理CUDA和GPU信息）：

   sudo apt install nvidia-cuda-toolkit

3. 查看CUDA版本：

   nvidia-smi

这将显示你的CUDA版本和其他相关信息。

然后，查看PyTorch版本：

1. 首先需要安装Python如果尚未安装：

   sudo apt install python3-pip

2. 使用pip安装`torch`库并查看版本：

   pip3 install torch torchvision

安装完成后，可以运行：

   import torch
   print(torch.__version__)

这将显示你当前安装的PyTorch库的版本。