网站优化方法页面重庆seo搜索引擎优化推荐
一、通过python查看
import torch
print(torch.cuda.is_available())

二、通过 pip list 查看
在conda环境


可以看出torch 后面是2.1.4 + cu124 说明GPU环境安装成功。
如果是cpu环境,则是:

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

在conda环境


可以看出torch 后面是2.1.4 + cu124 说明GPU环境安装成功。
如果是cpu环境,则是:
