Ubuntu上的PyTorch可以用于深度学习。PyTorch是专为深度学习设计的开源框架,支持CPU和GPU计算,适用于图像分类、自然语言处理等各类深度学习任务。
安装时,可根据需求选择CPU或GPU版本:
安装后,可通过import torch和print(torch.cuda.is_available())验证是否安装成功及是否支持GPU[1,2,3,4,5,6,7,8,9,10]。
import torch
print(torch.cuda.is_available())