python深度学习没法调用gpu

问题遇到的现象和发生背景

在跑深度学习的时候cpu占用率到100%，查询了资料后，需要安装cuda。
现在安装好了，
import torch
print(torch.cuda.is_available())
的输出也为true。
但独立显卡gpu0依旧没有调用，集显gpu1使用率上升，但不高，搜索后使用
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
依旧没用。

我的解答思路和尝试过的方法

我想要达到的结果

cpu使用率下降，独显gpu0正确调用

训练的时候模型，数据都要to.device('cuda')

可以先确定使用的设备device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
之后模型和数据都用这个设备model = model.to(device)，data = data.to(device)