如何查看自己的torch和cuda版本是否配对

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

如果输出结果是False,说明当前pytorch和cuda版本不配对
如果输出结果为ture,则配对

如果程序中出现报错:Torch not compiled with CUDA enabled
也是torch 和cuda版本不配对的问题,解决问题期间可以用上述代码测试新安装的版本是否配对.显示True时,说明已配对,在运行程序多半就不会出现这个报错了.

你可能感兴趣的:(毕业设计,cuda,gpu,pytorch)