可以的，PyTorch 完全支持在 Linux 上运行，而且 Linux 其实是 PyTorch 官方支持最好、使用最广泛的平台之一。
1. CPU 版本
pip install torch torchvision
2. GPU 版本（CUDA）
pip install torch torchvision --index-url https://download.pytorch.org/whl/cu118
3. Conda 安装
conda install pytorch torchvision pytorch-cuda=11.8 -c pytorch -c nvidia
import torch
print(torch.__version__)
print(torch.cuda.is_available()) # GPU 是否可用
如果你愿意，可以告诉我：
我可以帮你给出最合适的安装命令。
