这个问题在不同的运行环境下有不同的解决方案。
一种可行的解决方案:在代码中设置CUDA_VISIBLE_DEVICES
环境变量后,调用torch.cuda.device_count.cache_clear()
,例如:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "1"
import torch
torch.cuda.device_count.cache_clear()
这个问题在不同的运行环境下有不同的解决方案。
一种可行的解决方案:在代码中设置CUDA_VISIBLE_DEVICES
环境变量后,调用torch.cuda.device_count.cache_clear()
,例如:
import os
os.environ['CUDA_VISIBLE_DEVICES'] = "1"
import torch
torch.cuda.device_count.cache_clear()