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是不要加--recursive,因为国内镜像站不支持。
2、cd pytorch
3、git checkout v1.7.0
4、修改第三方库的网址
因为第1步没有加--recursive,这里需要手动修改pytorch文件夹下的.gitmodules文件,将里面所有https://github.com的网址都替换成https://github.com.cnpmjs.org,以确保网络不会在clone过程中中断。除了pytorch根目录下的.gitmodules文件外,还有这些地方也存在.gitmodules文件,需要一并修改&#