在运行代码时，发现指定了gpu，但是仍然使用的时gpu0，gpu0已经占满，导致
解决措施：一定要在import torch前加。
但是要注意gpu编号发生了变化，原来的gpu2编号为0
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '2,3'
import torch
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
