AssertionError: Torch not compiled with CUDA enabled

Pytorch和CUDA版本不兼容,运行python后(终端输入python回车)用以下代码测试

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

返回False则说明目前的pytorch版本无法使用显卡,如下图所示

AssertionError: Torch not compiled with CUDA enabled_第1张图片

接着重装合适版本的pytorch。。。 

你可能感兴趣的:(人工智能,深度学习)