windows10 编译pytorch源码

本文主要是介绍windows10 编译pytorch源码，对大家解决编程问题具有一定的参考价值，需要的程序猿们随着小编来一起学习吧！

之前gtx1660显卡在高价位卖了，现在只能凑合用gtx650爷爷级显卡。但是使用yolov5时出现

PyTorch no longer supports this GPU because it is too old

百度后解释说gtx650算力太低，当前发布最新版本不支持其显卡。

所以之只能靠自己编译源码

首先下载pytorch源码  我是用1.8.0版本

git clone --recursive https://github.com/pytorch/pytorch
cd pytorch
# if you are updating an existing checkout
git submodule sync
git submodule update --init --recursive
 

根据github指示进行下载，这个最好翻墙节省时间。

再用conda建立环境，安装编译环境

conda install astunparse numpy ninja pyyaml mkl mkl-include setuptools cmake cffi typing_extensions future six requests dataclasses

conda install -c conda-forge libuv=1.39

这里面cmake安装时老版本的可以去掉，手动安装cmake最新版本

https://zhuanlan.zhihu.com/p/106977910

参考这篇文章使用vs2019进行编译，找到你对应显卡算力，我显卡为gtx650 对应算力为3.0

因此要设置环境 set TORCH_CUDA_ARCH_LIST=3.0

然后再修改代码，不然安装tourchversion会说CUDA_ARCH=3.0 没有找到

修改\pytorch\torch\utils\cpp_extension.py中

添加3.0，然后进行编译

python setup.py install

中途编译可能会出错，把杀毒软件啥的都关掉就行，win10编译很玄学。

后面就是安装编译tourchvision

下载代码 直接编译即可

花了两天时间解决编译问题，太麻烦。。

这篇关于windows10 编译pytorch源码的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！