深度学习
冷犬
这个作者很懒,什么都没留下…
展开
-
pytorch 代码出现 ‘segmentation fault (core dump)’ 问题
在上线服务时,在本机以及服务器上测试能跑的代码突然提示segmentation fault (core dump)搞了很久检查了一下线上机器的pytorch版本为1.2.0,而自己的机子上是1.1.0,于是尝试重装pytorch.pip install torch==1.1.0然后就神奇地解决了。...原创 2020-11-16 20:26:41 · 2431 阅读 · 0 评论 -
深度学习pipeline - 自动识别空闲GPU
# !bin/bash# 一些英伟达的命令nvidia-smi -h # 帮助nvidia-smi -q / nvidia-smi --query # 显示所有gpu的信息.nvidia-smi -q -d / nvidia-smi -q --display= # 只显示指定的信息: MEMORY, UTILIZATION,PIDS等.知道这些基本的英伟达的命令之后,我们就...原创 2020-01-10 17:12:33 · 903 阅读 · 2 评论