运行程序报错如下:
cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cp
我的是因为指定了几个GPU, 换了一个环境运行,没有那么多gpu, 将下面的注掉就解决问题了
os.environ['CUDA_VISIBLE_DEVICES'] = '1,4'
运行程序报错如下:
cuda runtime error (100) : no CUDA-capable device is detected at /pytorch/aten/src/THC/THCGeneral.cp
我的是因为指定了几个GPU, 换了一个环境运行,没有那么多gpu, 将下面的注掉就解决问题了
os.environ['CUDA_VISIBLE_DEVICES'] = '1,4'