- 博客(3)
- 收藏
- 关注
RSS订阅原创 WSL2 编译安装pytorch1.7.0
WSL2要使用多显卡分布式计算，只能安装NCCL 2.11.4及以上版本，因此需要通过源码编译安装pytorch1.7.0。本博客适用于连接github不畅的用户。主要参考了这篇文章，并有所改动一、clone源码1、git clone https://github.com.cnpmjs.org/pytorch/pytorch这里注意两点：1是国内访问github速度堪忧而且时断时续，所以clone使用国内镜像站的地址，我使用的是cnpmjs.org/，当然也可以用别的。2是不要加--recurs
2021-11-15 17:57:31
1412
1
原创 WSL2，NCCL报错RuntimeError: NCCL Error 2: unhandled system error
在WSL2上使用pytorch1.7.1，无法进行多显卡分布式训练，提示“RuntimeError: NCCL Error 2: unhandled system error”，一直不知道什么意思，后来上网搜索，发现可以打印nccl的日志（链接） ，以获得更为详细的出错提示。随即在环境变量里添加了export NCCL_DEBUG=infoexport NCCL_SOCKET_IFNAME=eth0export NCCL_IB_DISABLE=1再次训练，这次打印出了更为详细的错误信息。如下
2021-11-03 14:18:14
7413
1
原创 Siammask目标跟踪使用SiamMask_VOT.pth训练自己的模型
想用siammask训练自己的模型，又不想从头开始训练，希望能够在项目组提供的SiamMask_VOT.pth的基础上继续训练，网上搜了一下，似乎没有相关教程，于是自己尝试了一下，简单记录步骤如下：1、将siammask_sharp文件夹下的run.sh修改如下：if [ -z "$1" ]thenecho "Need input base model!"echo "Usage: bash `basename "$0"` \$BASE_MODEL"exitfiROOT=`git rev-p
2021-06-26 08:00:17
1302
12
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人