检查版本:
输入内容:
import torch
print(torch.__version__)
返回结果:实例:1.7.1
或者
import torch
torch.cuda.is_available()
返回True即为安装成功
检查版本:
输入内容:
import torch
print(torch.__version__)
返回结果:实例:1.7.1
或者
import torch
torch.cuda.is_available()
返回True即为安装成功