ubuntu安装pytorch-gpu版本

要在Ubuntu上安装PyTorch-GPU版本，可以按照以下步骤进行操作：

1. 首先，检查你的显卡是否支持CUDA。可以在终端中运行以下命令来查看：

   lspci | grep -i nvidia

2. 安装NVIDIA驱动程序。可以通过以下方式在终端中安装最新的NVIDIA驱动程序：

   sudo apt-get install nvidia-driver-<version>

其中，`<version>`是你所需的驱动程序版本号。

3. 安装CUDA Toolkit。可以按照以下步骤在终端中安装CUDA Toolkit：
- 访问NVIDIA的官方网站，下载适用于你的Ubuntu版本的CUDA Toolkit安装包。
- 运行下载的安装包并按照提示进行安装。

4. 创建和激活一个新的虚拟环境（可选）。可以使用conda或者virtualenv来创建一个新的虚拟环境，并激活它。

5. 安装PyTorch-GPU。可以使用conda或者pip来安装PyTorch-GPU。下面是使用conda安装的示例命令：

   conda install pytorch torchvision torchaudio cudatoolkit=<cuda_version> -c pytorch

其中，`<cuda_version>`是你安装的CUDA Toolkit的版本号。

以上是在Ubuntu上安装PyTorch-GPU版本的一般步骤。请根据你的具体情况进行操作。