pytorch2.0.1 编译

  1. conda create -n xxx
  2. git clone --recursive https://github.com/pytorch/pytorch.git -b v2.5.1 (如果ssh需要秘钥）
  3. git submodule sync
  4. git submodule update --init --recursive
  5. 开始编译
export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which conda))/../"}
MAX_JOBS=16 USE_NINJA=1 USE_CUDA=1 DEBUG=0 USE_ROCM=0 USE_DISTRIBUTED=0 USE_MKLDNN=0 BUILD_TEST=0 USE_FBGEMM=0 USE_NNPACK=0 USE_QNNPACK=0 USE_XNNPACK=0 python3  setup.py develop
(如果cmake报错，一般镜像问题)

#cp /usr/lib/x86_64-linux-gnu/libstdc++.so.6 /home/liyuan/anaconda3/envs/torch201/lib/libstdc++.so.6
  1. 测试
python 
import torch
torch.cuda.is_available()

更好的编译方法：

cd build
cmake -DCMAKE_BUILD_TYPE=Debug -DMAX_JOBS=64 -DUSE_NINJA=0 -DUSE_CUDA=0 -DDEBUG=1 -DUSE_ROCM=0 -DUSE_DISTRIBUTED=0 -DUSE_MKLDNN=0 -DBUILD_TEST=0 -DUSE_FBGEMM=0 -DUSE_NNPACK=0 -DUSE_QNNPACK=0 -DUSE_XNNPACK=0 ..
make -j64 && cp lib/* ../torch/lib/

查看编译选项

$ python -c "import torch;print(torch.__config__.show())"
posted @ 2025-01-19 13:59  恰逢其时40  阅读(82)  评论(0)    收藏  举报