可以的，Ubuntu 是完全可以用来训练 PyTorch 的，而且这是非常常见、甚至是最主流的选择之一。
下面给你简单说明一下：
在 Ubuntu 上训练 PyTorch，一般需要：
# 创建虚拟环境（推荐）
python -m venv venv
source venv/bin/activate
# 安装 PyTorch（以 CUDA 11.8 为例）
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
CPU 版则直接：
pip install torch torchvision torchaudio
Ubuntu 不仅能训练 PyTorch，而且是目前最推荐的系统之一。
如果你愿意，我可以：
你现在是 本地电脑 还是 服务器 / 云主机？有显卡吗？
