深度学习报错 | THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp

错误定位

近日在自己的服务器上跑别人的代码时，反复出现报错：
THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp)
具体报错情况为，把张量加载到设备cuda上时报错，即无法利用gpu进行计算和训练，初步推测应为 pytorch 和 cuda 版本不匹配。

解决历程

1. 如果是面对别人的代码，建议大家新开 test.py，单独测试代码能否运用
   gpu（我的结果是不行，代表原来的环境有问题），测试代码可参考如下：

import torch

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
print(device