linux怎么安装pytorch-gpu

安装 PyTorch-GPU 需要先安装 CUDA 和 cuDNN，然后使用 pip 安装 PyTorch-GPU。具体步骤如下：

1. 安装 CUDA 和 cuDNN。可以参考 NVIDIA 官方文档进行安装。

2. 创建一个虚拟环境（可选）。

3. 打开终端，进入虚拟环境（如果有的话）。

4. 使用以下命令安装 PyTorch-GPU：

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

注意，这里的 cu111 对应的是 CUDA 11.1 版本，如果你安装的是其他版本的 CUDA，需要相应地修改 cuXXX 的值。

5. 安装完成后，可以使用以下命令测试是否安装成功：

   python -c "import torch; print(torch.cuda.is_available())"

如果输出 True，则说明安装成功。