用pytorch跑实验需要用到cuda加速，于是乎开始了下面的操作(这也是看了pytorch的官方tutorial)

cuda_device = torch.device('cuda:1')

兴致勃勃的开始实验，但是出现了rt所述的错误，然后就进行各种google，但是网上的方法不适用（别人的方法），按照他们的说法是因为只有1张GPU卡所以出现了错误，但我的GPU超过一张，还是出错，刚开始以为是cuda版本不对呢，然后重装cuda，最后发现还是不行，无奈最后只能用最原始的方法了，在代码中指定要使用的GPU卡id，然后继续coding。。。
代码如下：

os.environ['CUDA_VISIBLE_DEVICES'] = '1'