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()
本文标题:pytorch和cuda的版本适配问题
本文链接:https://www.haomeiwen.com/subject/wsmfwhtx.html
网友评论