pytorch和cuda的版本适配问题
2020-04-28 本文已影响0人
亦珩
import torch
print(torch.__version__)#注意是双下划线
>>>import torch
>>>torch.version.cuda#输出一个 cuda 版本
torch.cuda.is_available()
import torch
print(torch.__version__)#注意是双下划线
>>>import torch
>>>torch.version.cuda#输出一个 cuda 版本
torch.cuda.is_available()