pytorch源码编译踩坑记录

为了拯救老GPU,只能用源码编译了。

我不喜欢用conda,依赖包pip安装

基本环境:win7x64,python3.6.2,vs2017community,cuda9.0

cudnn安装跟cuda版本适应的即可

大部分步骤直接按照github上的说明来

最重要的是装上vs2017的toolset版本14.11,从VS安装里面选择即可,没有这个编译不了

编译会产生10G的文件,安装倒是只有500M多,需要保证硬盘空间充足

pytorch说的不支持旧显卡,指的是编译好的包里面不包含对旧显卡的支持,源码编译不存在这个问题,只是在每次运行python环境中第一次调用torch.sensor.cuda()时会报一个warning。

    原文作者:东方硕
    原文地址: https://zhuanlan.zhihu.com/p/60267229
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞