pytorch 源码编译安装

pytorch 源码编译安装

支持cuda, cudnn

1. 安装好GPU dirver, CUDA, cuDNN (注意对应版本)

2. 下载pytorch 源码

git clone https://github.com/pytorch/pytorch.git
cd pytorch
git pull 
git submodule sync
git submodule update --init --recursive

3. 编译

export DEBUG=1
python setup.py bdist_wheel

4. 安装编译的whl 包

pip install dist/torch*.whl

5. 卸载

pip uninstall torch
python setup.py clean

https://blog.csdn.net/CSDN_ten/article/details/132636688
https://findhao.net/academic/2590
https://blog.csdn.net/qq_36755743/article/details/136877045
https://codle.net/install-pytorch-from-source/
https://bbs.huaweicloud.com/blogs/407977