jupyter 服务似乎挂掉了,但是会立即重启……怎么解决

2020-11-13 10:34:43 浏览数 (1)

主要原因是显存不足,可以使用

nvidia-smi命令查看GPU使用情况。详见https://blog.csdn.net/IT_flying625/article/details/103194767

然后寻找显存占用最少的GPU指定使用。

代码语言:javascript复制
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '1'

应该就没问题了,除非没有找到可用的GPU资源,或者模型太大。

0 人点赞