python运行错误:RuntimeError: CUDA error: invalid device ordinal

用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'

你可能感兴趣的:(Python)