ubuntu16.04安装pytorch

安装 PyTorch 在 Ubuntu 16.04 中需要几个步骤:

1. 安装依赖:

sudo apt-get update
sudo apt-get install -y python-pip python-dev

2. 通过 pip 安装 PyTorch:

pip install torch torchvision

3. 确认安装:

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

如果安装正确，你将看到 PyTorch 的版本号。

注意: 如果你需要 CUDA 支持,请确保你已经安装了 NVIDIA 驱动与 CUDA 库, 并在 pip 安装时指定对应版本的 PyTorch.

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