pytorch和cuda的版本适配问题

2020-04-28  本文已影响0人  亦珩

import torch

print(torch.__version__)#注意是双下划线

>>>import torch

>>>torch.version.cuda#输出一个 cuda 版本

torch.cuda.is_available()

上一篇 下一篇

猜你喜欢

热点阅读