ubuntu安装pytorch gpu版本

安装PyTorch GPU版本的步骤如下：
1. 确保你的电脑有NVIDIA GPU。
2. 安装CUDA和cuDNN，可以在NVIDIA官网下载安装。
3. 在终端中输入命令：

   pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

其中"cu111"是CUDA版本，可能需要更换。
4. 安装完成后，可以使用如下命令验证是否安装成功:

   python -c "import torch; print(torch.__version__)"

5. 如果你的电脑有多块显卡，可以在设置中设置使用的显卡。

注意：如果你的系统是ubuntu18.04或更高版本，请确保安装了nvidia-driver-450或更高版本。