在Linux上编译PyTorch需要遵循以下步骤:
sudo apt-get update
sudo apt-get install -y cmake git wget unzip yasm pkg-config libopenblas-dev liblapack-dev libjpeg-dev libpng-dev
sudo apt-get install -y python3 python3-pip python3-venv
创建一个新的虚拟环境并激活它:
python3 -m venv pytorch_env
source pytorch_env/bin/activate
pip install numpy typing_extensions future six requests dataclasses
git clone --recursive https://github.com/pytorch/pytorch
cd pytorch
export CMAKE_PREFIX_PATH=${CONDA_PREFIX:-"$(dirname $(which python))/../"}
export USE_CUDA=1 # 如果您想使用CUDA,请设置为1,否则设置为0
export USE_CUDNN=1 # 如果您想使用cuDNN,请设置为1,否则设置为0
export USE_MKLDNN=1 # 如果您想使用MKL-DNN,请设置为1,否则设置为0
export USE_NNPACK=1 # 如果您想使用NNPACK,请设置为1,否则设置为0
export BUILD_SHARED_LIBS=1 # 如果您想构建共享库,请设置为1,否则设置为0
python setup.py install
编译过程可能需要一段时间,具体取决于您的硬件配置。
python -c "import torch; print(torch.__version__)"
如果看到PyTorch的版本号输出,那么恭喜您,您已成功在Linux上编译了PyTorch!