广西执业药师培训网站佛山模板网站建设

查看：torch 

import torch# 输出带CPU，表示torch是CPU版本的
print(f'torch的版本是：{torch.__version__}')
#
print(f'torch是否能使用cuda：{torch.cuda.is_available()}')

修改一下代码，将cuda改成cpu

最后运行正常：