Ubuntu编译GPU版本Pytorch

Ubuntu编译GPU版本Pytorch

摘要

两年前，疫情爆发后，呆在家中，无奈在有限的环境下编译了GPU版本Pytorch，这是当时的总结。前面当时编出来的whl包和源码没有留存。讽刺的是，现在我再按照自己的总结编译Pytorch时，依然走了很多的弯路。先前的总结没有注意所编译的pytorch版本，这次需要编译特定版本，对第三方的库git clone时也令人崩溃。希望能给到同样需要编译pytorch的人一些帮助。

环境

操作系统：Ubuntu18.04
硬件环境：
CPU: i5 3210m
RAM: 6G
显卡: GT640m
显卡驱动：470.103.01
CUDA: cuda10.0
CC: gcc7.5
CXX: g++7.5
python: 3.7.4
cmake:3.13
clang: 6.0.0
需要编译的pytorch:1.6.0
注意：

• 显卡型号、cuda版本和cudnn版本三者需要配合。比如CUDA11.0不支持GT640m显卡。
• 新版本的Pytorch(v1.10.1)要求更高版本的CUDA&