问题遇到的现象和发生背景
在跑深度学习的时候cpu占用率到100%,查询了资料后,需要安装cuda。
现在安装好了,
import torch
print(torch.cuda.is_available())
的输出也为true。
但独立显卡gpu0依旧没有调用,集显gpu1使用率上升,但不高,搜索后使用
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
依旧没用。
我的解答思路和尝试过的方法
我想要达到的结果
cpu使用率下降,独显gpu0正确调用