检查 torch.cuda.is_available()
是否为 True
# 在终端进入虚拟python环境 直接一行行输入 每行输完记得回车
python
import torch
torch.__version__
torch.cuda.is_available()
检查 torch.cuda.is_available()
是否为 True
# 在终端进入虚拟python环境 直接一行行输入 每行输完记得回车
python
import torch
torch.__version__
torch.cuda.is_available()