pytorch2.0.1 编译

1. conda create -n xxx
2. git clone –recursive https://github.com/pytorch/pytorch (注意，不要使用之前的编译目录）
3. git checkout v2.0.1 (不能-b）
4. git submodule sync
5. git submodule update –init –recursive
6. 开始编译

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
#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()

来源链接：https://www.cnblogs.com/qfqs40/p/18679539