宜兴淘宝网站建设义乌网站建设软件开发

一、通过python查看

import torch
print(torch.cuda.is_available())

二、通过 pip list 查看

在conda环境

可以看出torch 后面是2.1.4 + cu124 说明GPU环境安装成功。

如果是cpu环境，则是：