PyTorch版本检测及GPU是否安装成功检测

PyTorch GPU版本安装方法,请看anaconda虚拟环境下镜像安装PyTorch,利用jupyter notebook访问虚拟环境

检测程序如下:

import torch

print(torch.__version__)
print(torch.cuda.is_available())

运行结果:

1.4.0

True

PyTorch版本为1.4.0
True代表GPU版本成功安装!

发布了9 篇原创文章 · 获赞 18 · 访问量 3069

猜你喜欢

转载自blog.csdn.net/qq_45032341/article/details/105196680