import LTP后无法通过os.environ指定使用的显卡

[huhk-sysu]

问题

在使用from ltp import LTP后，无法通过os.environ['CUDA_VISIBLE_DEVICES']指定使用的显卡，默认使用0号卡。

版本信息

使用的Pytorch版本：1.2.0 （由于cuda版本受限无法在更高版本上实验） 尝试使用的ltp版本：4.0.9 / 4.1.2 （均有问题）

复现步骤：

出问题的情况

import os
# 先import LTP，再修改
from ltp import LTP
os.environ['CUDA_VISIBLE_DEVICES'] = "1"
import torch
# 执行下一步前，程序尚未占用显存
torch.tensor([1.0]).cuda()
# 执行此步后，使用nvidia-smi查看显存状况，发现python错误地使用了0号显卡

正常情况

import os
# 先修改，再import LTP
os.environ['CUDA_VISIBLE_DEVICES'] = "1"
from ltp import LTP
import torch
# 执行下一步前，程序尚未占用显存
torch.tensor([1.0]).cuda()
# 执行此步后，使用nvidia-smi查看显存状况，发现python正常使用了1号显卡