故事的开头还要从下面的错误说起:
AssertionError: Torch not compiled with CUDA enabled
网上查了一下大概的意思就是Pytorch和CUDA版本不兼容。
输入以下指令
import torch
print(torch.__version__)
print(torch.cuda.is_available())
输出的结果是False,说明当前的Pytorch版本无法使用显卡。
按照网上所说,先检查一下当前的CUDA版本,在终端中输入
cat /usr/local/cuda