Pytorch
文章平均质量分 52
左边cs
这个作者很懒,什么都没留下…
展开
-
linux无进程占用GPU,但却显示显存占满
今天在使用pytorch训练模型时提示:RuntimeError: CUDA out of memory通过nvidia-smi指令查看了GPU的使用情况,结果很奇怪:只有几个基础进程占用了大概200M的GPU,但GPU的总体占用为:24162MiB / 23259 MiB由于无相关进程显示,所以没有办法通过杀进程的方式释放显存。通过 gpustat 指令查看了其他用户进程的GPU使用情况:gpustat -u --color发现和nvidia-smi显示的无区别, 其中root用户的X原创 2021-08-18 11:15:22 · 3336 阅读 · 0 评论 -
Pytorch 学习记录
Pytorch 学习记录DATASETS & DATALOADERSPytorch Tutorials 链接DatasetsDatasets类是Pytorch中存储数据的结构,也可以自己定制Datasets类,方法很简单,可参考上方的Pytorch Tutorials 链接。Datasets有以下特性,容易忘记,记录一下:Datasets类的对象支持索引:如dataset[0] 就是第0条数据(data0, label0),是一个turple; 其中data0 的类型为tensor;原创 2021-05-29 22:44:26 · 122 阅读 · 0 评论