![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
zxxRobot
这个作者很懒,什么都没留下…
展开
-
PyTorch训练可视化(一)——可视化工具 TensorboardX
1.安装 TensorboardXpip install tensorboardX==1.8 后面的==1.8 是版本号。tensorboardX运行时常会报一些奇奇怪怪的错误,如AttributeError: 'torch._C.Value' object has no attribute 'debugName'。这通常是版本号的问题。多试几个版本就可以了。2.数字 (sca...原创 2019-11-18 11:01:40 · 973 阅读 · 4 评论 -
GPU使用笔记 (pytorch)
1. 实时查看nvidia GPU使用情况(适用于Linux和Windows)$ nvidia-smi # 显示当前GPU使用情况$ watch -n 1 -d nvidia-smi # 每隔1s刷新一次,时间参数可以更改2.使用多GPU加速训练torch.cuda.is_available() 返回True表示有GPU。torch.cuda.device_coun...原创 2019-11-04 13:05:47 · 1542 阅读 · 1 评论