PyTorch可以直接在Ubuntu上使用,需先安装对应版本。以下是关键信息:
pip
conda
print(torch.__version__)
print(torch.cuda.is_available())
venv
具体安装命令可参考PyTorch官网https://pytorch.org/get-started/locally/,根据系统环境选择适配方案。