1. 检验pytorch是否成功安装
import torch
# 检验pytorch安装情况
print('Hello World {}'.format(torch.__version__))
# 检验GPU调用情况
print(torch.cuda.is_available())
结果展示:
证明pytorch已成功安装,并且可以调用GPU进行加速。
1. 检验pytorch是否成功安装
import torch
# 检验pytorch安装情况
print('Hello World {}'.format(torch.__version__))
# 检验GPU调用情况
print(torch.cuda.is_available())
结果展示:
证明pytorch已成功安装,并且可以调用GPU进行加速。