ubuntu安装pytorch-gpu版本

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

首先，检查你的显卡是否支持CUDA。可以在终端中运行以下命令来查看：
lspci | grep -i nvidia
登录后复制

安装NVIDIA驱动程序。可以通过以下方式在终端中安装最新的NVIDIA驱动程序：
sudo apt-get install nvidia-driver-<version>
登录后复制
其中，<version>是你所需的驱动程序版本号。

安装CUDA Toolkit。可以按照以下步骤在终端中安装CUDA Toolkit：

访问NVIDIA的官方网站，下载适用于你的Ubuntu版本的CUDA Toolkit安装包。
运行下载的安装包并按照提示进行安装。

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

安装PyTorch-GPU。可以使用conda或者pip来安装PyTorch-GPU。下面是使用conda安装的示例命令：
conda install pytorch torchvision torchaudio cudatoolkit=<cuda_version> -c pytorch
登录后复制
其中，<cuda_version>是你安装的CUDA Toolkit的版本号。

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