做网站有必要注册商标吗短视频新媒体推广
一、通过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环境,则是:
