Ubuntu编译GPU版本Pytorch

Ubuntu18.04编译GPU版PyTorch1.6.0

原创

已于 2022-05-08 01:19:27 修改 · 774 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #ubuntu #深度学习

于 2022-05-06 21:23:21 首次发布

本文记录了在Ubuntu18.04上编译GPU版PyTorch1.6.0的过程，包括环境配置、源码获取、编译步骤，特别强调了CUDA、GCC版本与PyTorch的兼容性问题，以及如何使用Clang编译器进行编译。

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
注意：