CUDA unknown error - this may be due to an incorrectly set up environment

运行mmaction2时报错

builtins.RuntimeError: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero.

1.~/.bashrc中添加export CUDA_VISIBLE_DEVICES=0 2.代码中加入

import os
os.environ['CUDA_VISIBLE_DEVICES'] =‘0’

3.重启服务器